Commit Graph

2 Commits (2acb278e1fd6a78275533a62bf62b6eb7303fc8d)

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