github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-15 03:41:39 +00:00
Code Issues Packages Projects Releases Wiki Activity
b51bfec357
ColossalAI/colossalai/tensor/d_tensor
History
YuliangLiu0306 29386a54e6
[DTensor] refactor CommSpec (#3034)
2023-03-08 10:45:31 +08:00
..
__init__.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
comm_spec.py [DTensor] refactor CommSpec (#3034) 2023-03-08 10:45:31 +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 CommSpec (#3034) 2023-03-08 10:45:31 +08:00