ColossalAI/colossalai/fx
Jiarui Fang 36824a304c
[Doc] add more doc for ColoTensor. (#1458)
2022-08-16 10:38:41 +08:00
..
codegen [NFC] global vars should be upper case (#1456) 2022-08-15 09:50:29 +08:00
passes [Doc] add more doc for ColoTensor. (#1458) 2022-08-16 10:38:41 +08:00
tracer [fx] patched torch.max and data movement operator (#1391) 2022-08-01 15:31:50 +08:00
__init__.py [fx] supported data-dependent control flow in model tracing (#1185) 2022-06-29 15:05:25 +08:00
proxy.py [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00