Commit Graph

5 Commits

Author SHA1 Message Date
ver217
c415240db6 [nvme] CPUAdam and HybridAdam support NVMe offload (#1360)
* impl nvme optimizer

* update cpu adam

* add unit test

* update hybrid adam

* update docstr

* add TODOs

* update CI

* fix CI

* fix CI

* fix CI path

* fix CI path

* fix CI path

* fix install tensornvme

* fix CI

* fix CI path

* fix CI env variables

* test CI

* test CI

* fix CI

* fix nvme optim __del__

* fix adam __del__

* fix nvme optim

* fix CI env variables

* fix nvme optim import

* test CI

* test CI

* fix CI
2022-07-26 17:25:24 +08:00
Frank Lee
069d6fdc84 [workflow] update 8-gpu test to use torch 1.11 (#1332) 2022-07-18 11:41:13 +08:00
Frank Lee
4d5dbf48a6 [workflow] fixed trigger condition for 8-gpu unit test (#1323) 2022-07-15 15:00:02 +08:00
Frank Lee
5a9d8ef4d5 [workflow] fixed 8-gpu test workflow (#1101) 2022-06-13 13:50:22 +08:00
Frank Lee
03e52ecba3 [workflow] added regular 8 GPU testing (#1099)
* [workflow] added regular 8 GPU testing

* polish workflow
2022-06-10 17:38:15 +08:00