InternLM/internlm/core
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
..
communication Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
context Feat/add runntime gpu test (#254) 2023-09-01 13:38:01 +08:00
scheduler Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
engine.py initial commit 2023-07-06 12:55:23 +08:00
gradient_handler.py initial commit 2023-07-06 12:55:23 +08:00
naive_amp.py Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
trainer.py fix(core/trainer.py): fix streaming train state load error (#247) 2023-08-29 18:47:21 +08:00