ColossalAI/colossalai/fx/codegen/__init__.py

2 lines
45 B
Python
Raw Normal View History

from .activation_checkpoint_codegen import *