Jiarui Fang 68dcd51d41 [Tensor] update ColoTensor torch_function (#822) ...

* Revert "[zero] add ZeroTensorShardStrategy (#793)"

This reverts commit 88759e289e.

* [gemini] set cpu memory capacity

* [log] local throughput collecting

* polish

* polish

* polish

* polish code

* polish

* polish code

* add a new tensor structure and override linear for it

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* [tensor] renaming and reorganize directory structure.

* rm useless dir

* polish

* polish

* [tensor] hander the function not wrapped

* polish

2022-04-21 14:25:27 +08:00

..

_ops

[tensor] reorganize files (#820)

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

__init__.py

[tensor] reorganize files (#820)

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

colo_tensor.py

[Tensor] update ColoTensor torch_function (#822)

2022-04-21 14:25:27 +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