ColossalAI/colossalai/gemini/tensor
Jiarui Fang ab962b9735
[gemini] a new tensor structure (#818)
* 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
2022-04-21 11:42:37 +08:00
..
_ops [gemini] a new tensor structure (#818) 2022-04-21 11:42:37 +08:00
__init__.py [gemini] a new tensor structure (#818) 2022-04-21 11:42:37 +08:00
api.py [gemini] a new tensor structure (#818) 2022-04-21 11:42:37 +08:00
stateful_tensor.py [gemini] a new tensor structure (#818) 2022-04-21 11:42:37 +08:00
utils.py [gemini] a new tensor structure (#818) 2022-04-21 11:42:37 +08:00