Zihao 20e255d4e8 MemStatsCollectorStatic (#1765)		2022-11-07 16:49:03 +08:00
..
amp	…
auto_parallel	[autoparallel] add conv metainfo class for auto parallel (#1796)	2022-11-07 16:15:35 +08:00
builder	…
cli	…
communication	…
context	…
device	…
engine	…
fx	…
gemini	MemStatsCollectorStatic (#1765)	2022-11-07 16:49:03 +08:00
kernel	[hotfix] polish flash attention (#1802)	2022-11-07 14:30:22 +08:00
logging	…
nn	MemStatsCollectorStatic (#1765)	2022-11-07 16:49:03 +08:00
pipeline	…
registry	…
tensor	[NFC] polish type hint for shape consistency (#1801)	2022-11-07 14:13:03 +08:00
testing	…
trainer	…
utils	…
zero	MemStatsCollectorStatic (#1765)	2022-11-07 16:49:03 +08:00
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…