[hotfix] remove potiential circle import (#1307)

* make it faster * [hotfix] remove circle import
2025-09-05 02:51:59 +00:00 · 2022-07-14 13:44:26 +08:00
parent 6f2f9eb214
commit 4165eabb1e
18 changed files with 16 additions and 101 deletions
--- a/docs/colossalai/colossalai.engine.ophooks.rst
+++ b/docs/colossalai/colossalai.engine.ophooks.rst
@@ -1,11 +1,11 @@
-colossalai.engine.ophooks
+colossalai.gemini.ophooks
 =========================

-.. automodule:: colossalai.engine.ophooks
+.. automodule:: colossalai.gemini.ophooks
   :members:


 .. toctree::
   :maxdepth: 2

-   colossalai.engine.ophooks.zero_hook
+   colossalai.gemini.ophooks.zero_hook
--- a/docs/colossalai/colossalai.engine.ophooks.zero_hook.rst
+++ b/docs/colossalai/colossalai.engine.ophooks.zero_hook.rst
@@ -1,5 +1,5 @@
-colossalai.engine.ophooks.zero\_hook
+colossalai.gemini.ophooks.zero\_hook
 ====================================

-.. automodule:: colossalai.engine.ophooks.zero_hook
+.. automodule:: colossalai.gemini.ophooks.zero_hook
   :members:
--- a/docs/colossalai/colossalai.engine.rst
+++ b/docs/colossalai/colossalai.engine.rst
@@ -8,5 +8,5 @@ colossalai.engine
   :maxdepth: 2

   colossalai.engine.gradient_handler
-   colossalai.engine.ophooks
+   colossalai.gemini.ophooks
   colossalai.engine.schedule