colossalai.engine.ophooks

colossalai.engine.ophooks.register_ophooks_recursively(module, ophook_list=None, name='')

Recursilvely register pre/post hooks for all submodules in the module in FWD and BWD.