Extract the Python APIs in the pt1 dir back to the root (llvm#3237)

archana-ramalingam · May 8, 2024 · a763f77 · a763f77
1 parent 1a020fb
commit a763f77
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/projects/pt1/python/torch_mlir/torchscript.py b/projects/pt1/python/torch_mlir/torchscript.py
@@ -394,4 +394,4 @@ def compile(
         enable_ir_printing=enable_ir_printing,
     )
 
-    return lower_mlir_module(verbose, output_type, mb.module)
+    return lower_mlir_module(verbose, output_type, mb.module)
diff --git a/projects/pt1/python/torch_mlir_e2e_test/onnx_backends/linalg_on_tensors.py b/projects/pt1/python/torch_mlir_e2e_test/onnx_backends/linalg_on_tensors.py
@@ -77,4 +77,4 @@ def compile(self, imported_module: Module):
 
     def load(self, module):
         """Loads a compiled artifact into the runtime."""
-        return self.refbackend.load(module)
+        return self.refbackend.load(module)
diff --git a/python/torch_mlir/compiler_utils.py b/python/torch_mlir/compiler_utils.py
@@ -170,4 +170,4 @@ def lower_mlir_module(verbose, output_type, module):
             print("StableHLO Backend IR")
             print(module)
         return module
-    raise Exception(f"Unknown OutputType: {output_type}")
+    raise Exception(f"Unknown OutputType: {output_type}")