ColossalAI/tests/test_shardformer/test_layer
Frank Lee 3893fa1a8d [shardformer] refactored embedding and dropout to parallel module (#4013)
* [shardformer] refactored embedding and dropout to parallel module

* polish code
2023-07-04 16:05:01 +08:00
..
test_dropout.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00
test_embedding.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00
test_linear_1d.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00