mirror of https://github.com/hpcaitech/ColossalAI
ac863a01d6
* add benchmark * merge common func * add total and avg tflops Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com> |
||
---|---|---|
.. | ||
middleware | ||
rpc | ||
__init__.py | ||
layer_spec.py | ||
pipelinable.py | ||
pipeline_process_group.py | ||
utils.py |