ColossalAI/colossalai/fx/tracer
Frank Lee 7d6293927f
[fx] patched torch.max and data movement operator (#1391)
* [fx] patched torch.max and data movement operator

* polish code
2022-08-01 15:31:50 +08:00
..
meta_patch [fx] patched torch.max and data movement operator (#1391) 2022-08-01 15:31:50 +08:00
__init__.py
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
tracer.py [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00