github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ea0b52c12ee7598fe126ed0d8b0557f7e8a0e999
ColossalAI/colossalai/tensor/d_tensor
History
YuliangLiu0306 cd2b0eaa8d [DTensor] refactor sharding spec (#2987) 
* [autoparallel] refactor sharding spec

* rename function name
2023-03-07 11:08:11 +08:00
..
__init__.py
[DTensor] refactor sharding spec (#2987)
2023-03-07 11:08:11 +08:00
d_tensor.py
[DTensor] implementation of dtensor (#2946)
2023-03-01 16:34:58 +08:00
layout.py
[DTensor] refactor sharding spec (#2987)
2023-03-07 11:08:11 +08:00
misc.py
[DTensor] refactor sharding spec (#2987)
2023-03-07 11:08:11 +08:00
sharding_spec.py
[DTensor] refactor sharding spec (#2987)
2023-03-07 11:08:11 +08:00