mirror of https://github.com/hpcaitech/ColossalAI
7f2d2b2b5b
* [engine] fixed empty op hook check * polish code |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _memtracer_ophook.py | ||
| _shard_grad_ophook.py | ||
| _shard_param_ophook.py | ||
| utils.py | ||