ColossalAI/colossalai/fx/tracer
YuliangLiu0306 d87baa85d9
[autoparallel] support linear function bias addition (#2104)
2022-12-09 10:31:36 +08:00
..
bias_addition_patch [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00
meta_patch [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
__init__.py [fx] add a symbolic_trace api. (#1812) 2022-11-08 13:59:20 +08:00
_meta_trace.py [fx] refactor memory utils and extend shard utils. (#1754) 2022-10-26 14:24:41 +08:00
_symbolic_trace.py [fx] add more meta_registry for MetaTensor execution. (#2000) 2022-11-23 10:55:46 +08:00
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
experimental.py [fx] An experimental version of ColoTracer.' (#2002) 2022-12-07 18:36:17 +08:00
registry.py [autoparallel] support addbmm computation (#2102) 2022-12-08 21:15:11 +08:00
tracer.py [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00