ColossalAI/colossalai/tensor/d_tensor
Frank Lee 45d9384346 [shardformer] removed inplace tensor sharding (#4018) 2023-07-04 16:05:01 +08:00
..
RAEDME.md [shardformer] fix an error in readme (#3988) 2023-07-04 16:05:01 +08:00
__init__.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
api.py [shardformer] removed inplace tensor sharding (#4018) 2023-07-04 16:05:01 +08:00
comm_spec.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
d_tensor.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
layout.py [shardformer] integrated linear 1D with dtensor (#3996) 2023-07-04 16:05:01 +08:00
layout_converter.py [shardformer] integrated linear 1D with dtensor (#3996) 2023-07-04 16:05:01 +08:00
misc.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
sharding_spec.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
utils.py [DTensor] implement layout converter (#3055) 2023-03-10 09:53:52 +08:00