diff --git a/mlir/tools/mlir-cpu-runner/CMakeLists.txt b/mlir/tools/mlir-cpu-runner/CMakeLists.txt --- a/mlir/tools/mlir-cpu-runner/CMakeLists.txt +++ b/mlir/tools/mlir-cpu-runner/CMakeLists.txt @@ -17,8 +17,7 @@ MLIRJitRunner MLIRLLVMIR MLIRLLVMToLLVMIRTranslation - MLIROpenMP - MLIROpenMPToLLVMIRTranslation + MLIRToLLVMIRTranslationRegistration MLIRParser MLIRTargetLLVMIRExport MLIRSupport diff --git a/mlir/tools/mlir-cpu-runner/mlir-cpu-runner.cpp b/mlir/tools/mlir-cpu-runner/mlir-cpu-runner.cpp --- a/mlir/tools/mlir-cpu-runner/mlir-cpu-runner.cpp +++ b/mlir/tools/mlir-cpu-runner/mlir-cpu-runner.cpp @@ -16,8 +16,7 @@ #include "mlir/ExecutionEngine/JitRunner.h" #include "mlir/ExecutionEngine/OptUtils.h" #include "mlir/IR/Dialect.h" -#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h" -#include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.h" +#include "mlir/Target/LLVMIR/Dialect/All.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/TargetSelect.h" @@ -30,8 +29,7 @@ mlir::initializeLLVMPasses(); mlir::DialectRegistry registry; - mlir::registerLLVMDialectTranslation(registry); - mlir::registerOpenMPDialectTranslation(registry); + mlir::registerAllToLLVMIRTranslations(registry); return mlir::JitRunnerMain(argc, argv, registry); }