ColossalAI/colossalai/fx
YuliangLiu0306 fea3cb661c
[autoparallel] support addmm in tracer and solver (#1961)
* [fx] patch addmm

* [autoparallel] support addmm in tracer and solver
2022-11-16 14:59:18 +08:00
..
codegen
passes [SC] add GPT example for auto checkpoint (#1889) 2022-11-11 23:17:25 +08:00
profiler [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
tracer [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
__init__.py [fx] metainfo_trace as an API. (#1873) 2022-11-10 20:58:37 +08:00
_compatibility.py
_meta_registrations.py
graph_module.py
proxy.py