github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/cli/benchmark
History
YuliangLiu0306 de2f581d43
[cli] added micro benchmarking for tp (#789) 
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [CLI]add cli benchmark feature

* fix CodeFactor issues.

* refactor the module structure.
 		2022-04-19 12:08:28 +08:00
..
__init__.py [cli] added micro benchmarking for tp (#789) 2022-04-19 12:08:28 +08:00
run.py [cli] added micro benchmarking for tp (#789) 2022-04-19 12:08:28 +08:00
simple_model.py [cli] added micro benchmarking for tp (#789) 2022-04-19 12:08:28 +08:00
utils.py [cli] added micro benchmarking for tp (#789) 2022-04-19 12:08:28 +08:00