ColossalAI/colossalai/pipeline
Ziyue Jiang ac863a01d6
[example] add benchmark (#2276)
* add benchmark

* merge common func

* add total and avg tflops

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
2023-01-03 17:20:59 +08:00
..
middleware
rpc [example] add benchmark (#2276) 2023-01-03 17:20:59 +08:00
__init__.py
layer_spec.py
pipelinable.py
pipeline_process_group.py
utils.py