Skip to content

GitLab

"tests/test_zero/test_sharded_optim_v2.py" did not exist on "74f77e314baaab202f155263dbd13353bc89a04f"
Switch branch/tag
  1. 21 Apr, 2022 3 commits
    • Jiarui Fang's avatar
      [tensor] lazy init (#823) · 2ecc3d7a
      Jiarui Fang authored
      2ecc3d7a
    • Jiarui Fang's avatar
      [Tensor] apply ColoTensor on Torch functions (#821) · 660d2d1f
      Jiarui Fang authored 
      * Revert "[zero] add ZeroTensorShardStrategy (#793)"

This reverts commit 88759e28.

* [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
      660d2d1f
    • Jiarui Fang's avatar
      [tensor] reorganize files (#820) · 0ce8924c
      Jiarui Fang authored
      0ce8924c