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
..
amp	…
builder	…
cli	…
communication	…
context	…
engine	…
fx	[fx] patched torch.max and data movement operator (#1391)	2022-08-01 15:31:50 +08:00
gemini	[hotfix] fix a running error in test_colo_checkpoint.py (#1387)	2022-07-29 15:58:06 +08:00
kernel	…
logging	…
nn	[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)	2022-07-29 19:33:24 +08:00
pipeline	…
registry	…
tensor	[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)	2022-07-29 19:33:24 +08:00
testing	…
trainer	…
utils	[hotfix] fix a running error in test_colo_checkpoint.py (#1387)	2022-07-29 15:58:06 +08:00
zero	…
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…