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
Frank Lee 7d6293927f
[fx] patched torch.max and data movement operator (#1391)
2 years ago
..
codegen [fx] fixed indentation error in checkpointing codegen (#1385) 2 years ago
passes [hotfix] fix some bugs during gpt2 testing (#1379) 2 years ago
tracer [fx] patched torch.max and data movement operator (#1391) 2 years ago
__init__.py [fx] supported data-dependent control flow in model tracing (#1185) 2 years ago
proxy.py [doc] update rst and docstring (#1351) 2 years ago