Jiarui Fang 9bcd2fd4af [tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

..

__init__.py

[refactory] add nn.parallel module (#1068)

2022-06-06 15:34:41 +08:00

colo_module.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

embedding.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

linear.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

module_utils.py

[refactor] move process group from _DistSpec to ColoTensor. (#1203)

2022-07-06 16:15:16 +08:00