You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/fx/tracer
YuliangLiu0306 f6032ddb17
[autoparallel] fix bias addition module (#1800)
2 years ago
..
bias_addition_patch [autoparallel] fix bias addition module (#1800) 2 years ago
meta_patch [autoparallel] fix bias addition module (#1800) 2 years ago
__init__.py [fx] add a symbolic_trace api. (#1812) 2 years ago
_meta_trace.py [fx] refactor memory utils and extend shard utils. (#1754) 2 years ago
_symbolic_trace.py [fx] add a symbolic_trace api. (#1812) 2 years ago
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2 years ago
registry.py [fx] support module with bias addition (#1780) 2 years ago
tracer.py [autoparallel] add getattr handler (#1767) 2 years ago