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
tracer.py