Files
ColossalAI/colossalai
Hongxin Liu 646b3c5a90 [shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084)
* [tp] hotfix linear row

* [tp] support uneven split for fused linear

* [tp] support sp for fused linear

* [tp] fix gpt2 mlp policy

* [tp] fix gather fused and add fused linear row
2024-10-10 14:34:45 +08:00
..
2024-06-03 15:26:01 +08:00
2024-06-07 09:09:29 +00:00
2024-08-06 16:29:37 +08:00
2024-06-03 15:26:01 +08:00
2024-06-05 15:04:22 +08:00