ColossalAI/colossalai/fx
YuliangLiu0306 d2fc067231
[autoparallel] fix param hook issue in transform pass (#1755)
2022-10-24 13:13:38 +08:00
..
codegen
passes [autoparallel] fix param hook issue in transform pass (#1755) 2022-10-24 13:13:38 +08:00
profiler
tracer [autoparallel] added addbmm handler (#1751) 2022-10-21 18:55:48 +08:00
__init__.py
_compatibility.py
_meta_registrations.py
graph_module.py
proxy.py