diff --git a/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp b/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp --- a/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp +++ b/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp @@ -138,7 +138,7 @@ mlir::initializeLLVMPasses(); mlir::JitRunnerConfig jitRunnerConfig; - jitRunnerConfig.mlirTransformer = &runMLIRPasses; + jitRunnerConfig.mlirTransformer = runMLIRPasses; return mlir::JitRunnerMain(argc, argv, jitRunnerConfig); } diff --git a/mlir/tools/mlir-rocm-runner/mlir-rocm-runner.cpp b/mlir/tools/mlir-rocm-runner/mlir-rocm-runner.cpp --- a/mlir/tools/mlir-rocm-runner/mlir-rocm-runner.cpp +++ b/mlir/tools/mlir-rocm-runner/mlir-rocm-runner.cpp @@ -334,5 +334,9 @@ LLVMInitializeAMDGPUAsmPrinter(); mlir::initializeLLVMPasses(); - return mlir::JitRunnerMain(argc, argv, &runMLIRPasses); + + mlir::JitRunnerConfig jitRunnerConfig; + jitRunnerConfig.mlirTransformer = runMLIRPasses; + + return mlir::JitRunnerMain(argc, argv, jitRunnerConfig); } diff --git a/mlir/tools/mlir-spirv-cpu-runner/mlir-spirv-cpu-runner.cpp b/mlir/tools/mlir-spirv-cpu-runner/mlir-spirv-cpu-runner.cpp --- a/mlir/tools/mlir-spirv-cpu-runner/mlir-spirv-cpu-runner.cpp +++ b/mlir/tools/mlir-spirv-cpu-runner/mlir-spirv-cpu-runner.cpp @@ -88,8 +88,8 @@ mlir::JitRunnerConfig jitRunnerConfig; - jitRunnerConfig.mlirTransformer = &runMLIRPasses; - jitRunnerConfig.llvmModuleBuilder = &convertMLIRModule; + jitRunnerConfig.mlirTransformer = runMLIRPasses; + jitRunnerConfig.llvmModuleBuilder = convertMLIRModule; return mlir::JitRunnerMain(argc, argv, jitRunnerConfig); } diff --git a/mlir/tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp b/mlir/tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp --- a/mlir/tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp +++ b/mlir/tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp @@ -59,7 +59,7 @@ mlir::initializeLLVMPasses(); mlir::JitRunnerConfig jitRunnerConfig; - jitRunnerConfig.mlirTransformer = &runMLIRPasses; + jitRunnerConfig.mlirTransformer = runMLIRPasses; return mlir::JitRunnerMain(argc, argv, jitRunnerConfig); }