diff --git a/externals/llvm-project b/externals/llvm-project index 889c6f399..3580daacf 160000 --- a/externals/llvm-project +++ b/externals/llvm-project @@ -1 +1 @@ -Subproject commit 889c6f3996769a991a24da957f597e7500d158e7 +Subproject commit 3580daacf3ecb9bc628d05a4037266df6532ad6e diff --git a/include/torch-mlir-c/Dialects.h b/include/torch-mlir-c/Dialects.h index 20ded3520..99156c170 100644 --- a/include/torch-mlir-c/Dialects.h +++ b/include/torch-mlir-c/Dialects.h @@ -10,7 +10,7 @@ #ifndef TORCHMLIR_C_DIALECTS_H #define TORCHMLIR_C_DIALECTS_H -#include "mlir-c/Registration.h" +#include "mlir-c/IR.h" #ifdef __cplusplus extern "C" { diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt index 0a311c385..77a31184c 100644 --- a/python/CMakeLists.txt +++ b/python/CMakeLists.txt @@ -110,7 +110,7 @@ set(_source_components # tree, which seems excessive. MLIRPythonSources MLIRPythonExtension.Core - MLIRPythonExtension.AllPassesRegistration + MLIRPythonExtension.RegisterEverything MLIRPythonExtension.ExecutionEngine TorchMLIRPythonSources TorchMLIRPythonExtensions diff --git a/python/TorchMLIRModule.cpp b/python/TorchMLIRModule.cpp index c84f3b6f7..e0b045143 100644 --- a/python/TorchMLIRModule.cpp +++ b/python/TorchMLIRModule.cpp @@ -8,7 +8,6 @@ //===----------------------------------------------------------------------===// #include "mlir-c/Bindings/Python/Interop.h" -#include "mlir-c/Registration.h" #include "mlir/Bindings/Python/PybindAdaptors.h" #include "torch-mlir-c/Dialects.h" #include "torch-mlir-c/Registration.h" diff --git a/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/shape_lib_gen.py b/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/shape_lib_gen.py index aa0031324..8de260f2a 100644 --- a/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/shape_lib_gen.py +++ b/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/shape_lib_gen.py @@ -18,7 +18,6 @@ import torch.jit._shape_functions as upstream_shape_functions from torch_mlir.dialects.torch.importer.jit_ir import ModuleBuilder from torch_mlir.passmanager import PassManager -import torch_mlir.all_passes_registration from .registry import Registry diff --git a/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/module_builder.cpp b/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/module_builder.cpp index 70ac687b9..c5665ea9e 100644 --- a/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/module_builder.cpp +++ b/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/module_builder.cpp @@ -17,7 +17,6 @@ #include "mlir-c/BuiltinAttributes.h" #include "mlir-c/BuiltinTypes.h" #include "mlir-c/Diagnostics.h" -#include "mlir-c/Registration.h" #include "torch-mlir-c/Registration.h" namespace py = pybind11; diff --git a/python/torch_mlir_e2e_test/linalg_on_tensors_backends/refbackend.py b/python/torch_mlir_e2e_test/linalg_on_tensors_backends/refbackend.py index fd4110afe..8791f3b32 100644 --- a/python/torch_mlir_e2e_test/linalg_on_tensors_backends/refbackend.py +++ b/python/torch_mlir_e2e_test/linalg_on_tensors_backends/refbackend.py @@ -10,8 +10,6 @@ from torch_mlir.ir import * from torch_mlir.passmanager import * from torch_mlir.execution_engine import * from torch_mlir.runtime import * -# Imported for side effects. -import torch_mlir.all_passes_registration import torch_mlir.dialects.torch from torch_mlir.compiler_utils import run_pipeline_with_repro_report diff --git a/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py b/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py index eb98371fb..eefb492f0 100644 --- a/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py +++ b/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py @@ -5,8 +5,6 @@ from torch_mlir.ir import * from torch_mlir.passmanager import * -# Imported for side effects. -import torch_mlir.all_passes_registration from torch_mlir.compiler_utils import run_pipeline_with_repro_report from torch_mlir_e2e_test.linalg_on_tensors_backends.refbackend import RefBackendLinalgOnTensorsBackend