tests/test_shardformer/test_layer - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	test_dist_crossentropy.py	1.7 KB
	test_dropout.py	2.2 KB
	test_embedding.py	1.7 KB
	test_gpt2_qkv_fused_linear_1d.py	4.7 KB
	test_layernorm.py	1.5 KB
	test_linear_1d.py	10.2 KB
	test_qkv_fused_linear_1d.py	4.6 KB
	test_ring_attn.py	7.0 KB
	test_sequence_parallel.py	6.2 KB
	test_vocab_parallel_embedding_1d.py	1.8 KB