github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'feature/zerobubble'
${ noResults }
ColossalAI/tests/test_shardformer/test_layer
History
duanjunwen 1d328ff651
Merge branch 'main' into dev/zero_bubble 		3 weeks ago
..
test_dist_crossentropy.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago
test_dropout.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago
test_embedding.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago
test_gpt2_qkv_fused_linear_1d.py [shardformer] optimize seq parallelism (#6086) 2 months ago
test_layernorm.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago
test_linear_1d.py [fix] fix linear (no tp) ops func name; 4 weeks ago
test_qkv_fused_linear_1d.py [shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084) 2 months ago
test_ring_attn.py fix 1 month ago
test_sequence_parallel.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago
test_vocab_parallel_embedding_1d.py [misc] refactor launch API and tensor constructor (#5666) 7 months ago