Commit Graph

5 Commits (80f1eb9a3619949ecc4311e2dadc8dac59712600)

Author SHA1 Message Date
zaglc 9b1b0c5c20 fix format problem 2023-09-14 17:03:36 +08:00
zaglc 31d2a2916d feat(fsdp) add mix-precision 2 2023-09-08 10:16:58 +08:00
zaglc 85c6ed6473 feat(fsdp): add training option for fsdp 2023-09-04 18:01:30 +08:00
Sun Peng 860de0aa46
Feat/add runntime gpu test (#254)
* feat: add gpu bench

* feat/add allreduce runtime bench

---------

Co-authored-by: sunpengsdu <sunpengsdu@gmail.com>
2023-09-01 13:38:01 +08:00
Sun Peng fa7337b37b initial commit 2023-07-06 12:55:23 +08:00