tests/test_tensor - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

41377 0 1 Python

	test_dtensor
	test_comm_spec_apply.py	7.2 KB
	test_mix_gather.py	11.5 KB
	test_padded_tensor.py	1.7 KB
	test_shape_consistency_apply.py	2.6 KB
	test_shape_consistency.py	5.9 KB
	test_sharding_spec.py	728 B