fix the correct path for primus modules import

luiza-amd · luiza-amd · commit f761e98ea1f4 · 2026-07-13T07:51:07.000-05:00
diff --git a/primus/backends/megatron/patches/ddp_overlap_compile_patches.py b/primus/backends/megatron/patches/ddp_overlap_compile_patches.py
@@ -23,7 +23,7 @@
 import torch
 
 from primus.core.patches import PatchContext, get_args, register_patch
-from primus.modules.module_utils import log_rank_0
+from primus.core.utils.module_utils import log_rank_0
 
 
 @register_patch(
diff --git a/primus/backends/megatron/patches/torch_compile_patches.py b/primus/backends/megatron/patches/torch_compile_patches.py
@@ -12,7 +12,7 @@
 """
 
 from primus.core.patches import PatchContext, get_args, register_patch
-from primus.modules.module_utils import log_rank_0
+from primus.core.utils.module_utils import log_rank_0
 
 
 @register_patch(
