tests/test_device - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

41377 0 1 Python

	test_alpha_beta.py	982 B
	test_device_mesh.py	3.0 KB
	test_extract_alpha_beta.py	1.0 KB
	test_init_logical_pg.py	1.0 KB
	test_search_logical_device_mesh.py	1.0 KB