Commit Graph

2 Commits (b0219951994c8595720e572eec4ffbd53fbed04a)

Author SHA1 Message Date
Sun Peng 32664328e7
Feat/overlap_bcast_forward (#218)
* feat/support bcast forward overlao

* feat/optimize the bcast call

* feat/optimize the bcast call

* feat/optimize the bcast call

* fix lint

* fix lint

* fix lint

* fix lint

* add torch.cuda.synchronize in save_checkpoint

---------

Co-authored-by: sunpeng <sunpengsdu@gmail.com>
2023-08-23 16:59:59 +08:00
Sun Peng fa7337b37b initial commit 2023-07-06 12:55:23 +08:00