ColossalAI/tests/test_shardformer
Frank Lee 015af592f8 [shardformer] integrated linear 1D with dtensor (#3996)
* [shardformer] integrated linear 1D with dtensor

* polish code
2023-07-04 16:05:01 +08:00
..
test_layer [shardformer] integrated linear 1D with dtensor (#3996) 2023-07-04 16:05:01 +08:00
test_model [Shardformer] Downstream bert (#3979) 2023-07-04 16:05:01 +08:00
test_module [shardformer] Unit test (#3928) 2023-07-04 16:05:01 +08:00