fix

siliconflow · Apr 15, 2024 · 5fa9e9f · 5fa9e9f
1 parent c974d3a
commit 5fa9e9f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/onediff/infer_compiler/backends/nexfort.py b/src/onediff/infer_compiler/backends/nexfort.py
@@ -1,7 +1,6 @@
 import dataclasses
 import torch
 from .registry import register_backend
-from ..options import CompileOptions
 
 
 def make_inductor_options(options):
@@ -19,6 +18,7 @@ def compile(torch_module: torch.nn.Module, *, options=None):
     from nexfort.utils.memory_format import apply_memory_format
     from nexfort.compilers import nexfort_compile
     from ..nexfort.deployable_module import NexfortDeployableModule
+    from ..utils import CompileOptions
 
     options = options if options is not None else CompileOptions()
     nexfort_options = options.nexfort

diff --git a/src/onediff/infer_compiler/core/with_onediff_compile.py b/src/onediff/infer_compiler/core/with_onediff_compile.py
@@ -5,7 +5,7 @@
 def compile(
     torch_module: torch.nn.Module, *, backend="nexfort", options=None
 ) -> DeployableModule:
-    from .backends.registry import lookup_backend
+    from ..backends.registry import lookup_backend
 
     backend = lookup_backend(backend)
     model = backend(torch_module, options=options)

diff --git a/src/onediff/infer_compiler/nexfort/deployable_module.py b/src/onediff/infer_compiler/nexfort/deployable_module.py
@@ -1,5 +1,5 @@
 import torch
-from ..deployable_module import DeployableModule
+from ..core.deployable_module import DeployableModule
 
 
 class NexfortDeployableModule(DeployableModule):