tests/test_fp8 - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 1 Python

	test_all_to_all_single.py	2.5 KB
	test_fp8_all_to_all_single.py	1.2 KB
	test_fp8_all_to_all.py	1.5 KB
	test_fp8_allgather.py	1.5 KB
	test_fp8_allreduce.py	1.5 KB
	test_fp8_cast.py	1.0 KB
	test_fp8_ddp_comm_hook.py	2.5 KB
	test_fp8_fsdp_comm_hook.py	3.3 KB
	test_fp8_hook.py	1.4 KB
	test_fp8_linear.py	1.6 KB
	test_fp8_reduce_scatter.py	1.6 KB