mirror of https://github.com/InternLM/InternLM
![]() * feat: add gpu bench * feat/add allreduce runtime bench --------- Co-authored-by: sunpengsdu <sunpengsdu@gmail.com> |
||
---|---|---|
.. | ||
communication | ||
context | ||
scheduler | ||
__init__.py | ||
engine.py | ||
gradient_handler.py | ||
naive_amp.py | ||
trainer.py |