github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 09:24:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
dfca9678fa3fd35ef185370289a3aabffb8dbf85
ColossalAI/tests/test_shardformer/test_layer
History
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_linear_1d.py
[shardformer] integrated linear 1D with dtensor (#3996)
2023-07-04 16:05:01 +08:00