mirror of https://github.com/hpcaitech/ColossalAI
![]() * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages * [fx] modify the calculation of node_size in MetaInfoProp for activation checkpointing usages |
||
---|---|---|
.. | ||
codegen | ||
passes | ||
tracer | ||
__init__.py | ||
proxy.py |