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

..

__init__.py

[pipeline/pytree] add pytree to process args and kwargs | provide data_process_func to process args and kwargs after forward (#1642)

2022-09-29 10:58:58 +08:00

_pipeline_base.py

[example] add benchmark (#2276)

2023-01-03 17:20:59 +08:00

_pipeline_schedule.py

[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)

2022-12-23 11:38:43 +08:00

utils.py

fix type of num_worker_threads (#2237)

2022-12-30 11:04:01 +08:00