Fix (export): update torch-mlir jit_ir import path (#771)

Xilinx · Nov 28, 2023 · de165c3 · de165c3
1 parent 63370f4
commit de165c3
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/brevitas_examples/llm/llm_quant/mlir_custom_mm.py b/src/brevitas_examples/llm/llm_quant/mlir_custom_mm.py
@@ -9,10 +9,9 @@
 import torch
 import torch.utils.cpp_extension
 import torch_mlir
-from torch_mlir.dialects.torch.importer.jit_ir.build_tools.registry import \
-    _rename_python_keyword_parameter_name
-from torch_mlir.dialects.torch.importer.jit_ir.build_tools.registry import JitOperator
-from torch_mlir.dialects.torch.importer.jit_ir.build_tools.registry import SIG_ATTR_TYPE
+from torch_mlir.jit_ir_importer.build_tools.registry import _rename_python_keyword_parameter_name
+from torch_mlir.jit_ir_importer.build_tools.registry import JitOperator
+from torch_mlir.jit_ir_importer.build_tools.registry import SIG_ATTR_TYPE
 
 from brevitas.backport.fx._symbolic_trace import wrap