ColossalAI/colossalai/fx
Ziyue Jiang 59e343328d
[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)
* add splitter

* polish code

* remove comment

* fix async nan by moving to cpu first

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
2022-12-23 11:38:43 +08:00
..
codegen
passes [Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156) 2022-12-23 11:38:43 +08:00
profiler [hotfix] fix aten default bug (#2158) 2022-12-20 22:40:46 +08:00
tracer [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00
__init__.py
_compatibility.py
_meta_registrations.py register meta func for rnn (#2159) 2022-12-21 23:06:18 +08:00
graph_module.py
proxy.py