ColossalAI/colossalai/fx/tracer
YuliangLiu0306 e34e850a4c
[autoparallel]add essential CommActions for broadcast oprands (#1793)
2022-11-04 18:36:42 +08:00
..
bias_addition_patch [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
meta_patch [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
__init__.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
_meta_trace.py [fx] refactor memory utils and extend shard utils. (#1754) 2022-10-26 14:24:41 +08:00
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
registry.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
tracer.py [autoparallel] add getattr handler (#1767) 2022-11-03 12:31:33 +08:00