ColossalAI/colossalai
HELSON 02b058032d
[fx] meta registration compatibility (#3253)
* [fx] meta registration compatibility

* fix error
2023-03-27 15:22:17 +08:00
..
_C
_analyzer
amp
auto_parallel
autochunk
booster [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
builder
checkpoint_io [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
cli
cluster [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
communication
context
device
engine
fx [fx] meta registration compatibility (#3253) 2023-03-27 15:22:17 +08:00
gemini Add interface for colo tesnor dp size (#3227) 2023-03-27 09:42:21 +08:00
interface [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
kernel
logging
nn
pipeline
registry
tensor Add interface for colo tesnor dp size (#3227) 2023-03-27 09:42:21 +08:00
testing
trainer
utils [lazyinit] combine lazy tensor with dtensor (#3204) 2023-03-23 10:53:06 +08:00
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py