ColossalAI/colossalai/tensor at c6930d8ddfc5e288375da8d0bb3f6a97d7f3f708 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

YuliangLiu0306 c6930d8ddf [pipelinable]use ColoTensor to replace dummy tensor. (#853 )

2022-04-24 18:31:22 +08:00

..

[Tensor ] Add 1Drow weight reshard by spec (#854 )

2022-04-24 18:30:20 +08:00

__init__.py

[tensor] reorganize files (#820 )

2022-04-21 14:15:48 +08:00

colo_tensor.py

[pipelinable]use ColoTensor to replace dummy tensor. (#853 )

2022-04-24 18:31:22 +08:00

op_wrapper.py

[tensor] reorganize files (#820 )

2022-04-21 14:15:48 +08:00

utils.py

[tensor] reorganize files (#820 )

2022-04-21 14:15:48 +08:00