Skip to content

Commit dc9d2e5

Browse files
committed
[mlir] Add missing registrations to runners.
1 parent 16b192f commit dc9d2e5

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

mlir/tools/mlir-spirv-cpu-runner/mlir-spirv-cpu-runner.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
#include "mlir/ExecutionEngine/OptUtils.h"
2929
#include "mlir/Pass/Pass.h"
3030
#include "mlir/Pass/PassManager.h"
31+
#include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h"
3132
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
3233
#include "mlir/Target/LLVMIR/Export.h"
3334

@@ -111,6 +112,7 @@ int main(int argc, char **argv) {
111112
registry.insert<mlir::arith::ArithDialect, mlir::LLVM::LLVMDialect,
112113
mlir::gpu::GPUDialect, mlir::spirv::SPIRVDialect,
113114
mlir::func::FuncDialect, mlir::memref::MemRefDialect>();
115+
mlir::registerBuiltinDialectTranslation(registry);
114116
mlir::registerLLVMDialectTranslation(registry);
115117

116118
return mlir::JitRunnerMain(argc, argv, registry, jitRunnerConfig);

mlir/tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
#include "mlir/ExecutionEngine/JitRunner.h"
3535
#include "mlir/Pass/Pass.h"
3636
#include "mlir/Pass/PassManager.h"
37+
#include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h"
3738
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
3839
#include "llvm/Support/InitLLVM.h"
3940
#include "llvm/Support/TargetSelect.h"
@@ -114,6 +115,7 @@ int main(int argc, char **argv) {
114115
mlir::gpu::GPUDialect, mlir::spirv::SPIRVDialect,
115116
mlir::func::FuncDialect, mlir::memref::MemRefDialect,
116117
mlir::vector::VectorDialect>();
118+
mlir::registerBuiltinDialectTranslation(registry);
117119
mlir::registerLLVMDialectTranslation(registry);
118120

119121
return mlir::JitRunnerMain(argc, argv, registry, jitRunnerConfig);

0 commit comments

Comments
 (0)