Commit Graph

2 Commits (0e6b1f856cbc71b4926d975a55b0ae3e80a1d46d)

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