fix

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

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)

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):

src/onediff/infer_compiler/utils/env_var.py

@@ -33,13 +33,6 @@ def set_integer_env_var(env_var: str, val: Optional[int]):
 
 
 def _set_env_vars(field2env_var, options):
-    from .utils import (
-        parse_boolean_from_env,
-        set_boolean_env_var,
-        parse_integer_from_env,
-        set_integer_env_var,
-    )
-
     for field in dataclasses.fields(options):
         field_name = field.name
         field_value = getattr(options, field_name)