diff --git a/mlir/include/mlir/Support/JitRunner.h b/mlir/include/mlir/ExecutionEngine/JitRunner.h rename from mlir/include/mlir/Support/JitRunner.h rename to mlir/include/mlir/ExecutionEngine/JitRunner.h diff --git a/mlir/lib/ExecutionEngine/CMakeLists.txt b/mlir/lib/ExecutionEngine/CMakeLists.txt --- a/mlir/lib/ExecutionEngine/CMakeLists.txt +++ b/mlir/lib/ExecutionEngine/CMakeLists.txt @@ -6,6 +6,7 @@ ExecutionEngine.cpp RunnerUtils.cpp OptUtils.cpp + JitRunner.cpp ) add_mlir_library(MLIRExecutionEngine @@ -42,6 +43,31 @@ MLIRTargetLLVMIR ) +add_mlir_library(MLIRJitRunner + JitRunner.cpp + + EXCLUDE_FROM_LIBMLIR + + DEPENDS + intrinsics_gen + + LINK_COMPONENTS + Core + OrcJIT + JITLink + Support + + LINK_LIBS PUBLIC + MLIRExecutionEngine + MLIRIR + MLIRParser + MLIRStandardOps + MLIRTargetLLVMIR + MLIRTransforms + MLIRStandardToLLVM + MLIRSupport +) + add_mlir_library(mlir_c_runner_utils SHARED CRunnerUtils.cpp diff --git a/mlir/lib/Support/JitRunner.cpp b/mlir/lib/ExecutionEngine/JitRunner.cpp rename from mlir/lib/Support/JitRunner.cpp rename to mlir/lib/ExecutionEngine/JitRunner.cpp --- a/mlir/lib/Support/JitRunner.cpp +++ b/mlir/lib/ExecutionEngine/JitRunner.cpp @@ -14,7 +14,7 @@ // transformation. //===----------------------------------------------------------------------===// -#include "mlir/Support/JitRunner.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "mlir/Dialect/LLVMIR/LLVMDialect.h" #include "mlir/ExecutionEngine/ExecutionEngine.h" diff --git a/mlir/lib/Support/CMakeLists.txt b/mlir/lib/Support/CMakeLists.txt --- a/mlir/lib/Support/CMakeLists.txt +++ b/mlir/lib/Support/CMakeLists.txt @@ -1,6 +1,5 @@ set(LLVM_OPTIONAL_SOURCES FileUtilities.cpp - JitRunner.cpp MlirOptMain.cpp StorageUniquer.cpp ToolUtilities.cpp @@ -34,30 +33,3 @@ MLIRParser MLIRSupport ) - -# Exclude from libMLIR.so because the JIT infrastructure -# is a big dependency which most don't need. -add_mlir_library(MLIRJitRunner - JitRunner.cpp - - DEPENDS - intrinsics_gen - - EXCLUDE_FROM_LIBMLIR - - LINK_COMPONENTS - Core - OrcJIT - JITLink - Support - - LINK_LIBS PUBLIC - MLIRExecutionEngine - MLIRIR - MLIRParser - MLIRStandardOps - MLIRTargetLLVMIR - MLIRTransforms - MLIRStandardToLLVM - 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 @@ -13,7 +13,7 @@ //===----------------------------------------------------------------------===// #include "mlir/InitAllDialects.h" -#include "mlir/Support/JitRunner.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/TargetSelect.h" #include "mlir/ExecutionEngine/OptUtils.h" 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 @@ -28,7 +28,7 @@ #include "mlir/InitAllDialects.h" #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" -#include "mlir/Support/JitRunner.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "mlir/Transforms/DialectConversion.h" #include "mlir/Transforms/Passes.h" #include "llvm/Support/InitLLVM.h" 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 @@ -23,7 +23,7 @@ #include "mlir/InitAllDialects.h" #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" -#include "mlir/Support/JitRunner.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/TargetSelect.h"