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 @@ -12,11 +12,11 @@ // //===----------------------------------------------------------------------===// +#include "mlir/ExecutionEngine/JitRunner.h" +#include "mlir/ExecutionEngine/OptUtils.h" #include "mlir/InitAllDialects.h" -#include "mlir/Support/JitRunner.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/TargetSelect.h" -#include "mlir/ExecutionEngine/OptUtils.h" int main(int argc, char **argv) { mlir::registerAllDialects(); 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 @@ -22,13 +22,13 @@ #include "mlir/Dialect/GPU/Passes.h" #include "mlir/Dialect/LLVMIR/LLVMDialect.h" #include "mlir/Dialect/LLVMIR/NVVMDialect.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "mlir/ExecutionEngine/OptUtils.h" #include "mlir/IR/Function.h" #include "mlir/IR/Module.h" #include "mlir/InitAllDialects.h" #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" -#include "mlir/Support/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 @@ -19,11 +19,11 @@ #include "mlir/Dialect/GPU/Passes.h" #include "mlir/Dialect/SPIRV/Passes.h" #include "mlir/Dialect/SPIRV/SPIRVOps.h" +#include "mlir/ExecutionEngine/JitRunner.h" #include "mlir/ExecutionEngine/OptUtils.h" #include "mlir/InitAllDialects.h" #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" -#include "mlir/Support/JitRunner.h" #include "llvm/Support/InitLLVM.h" #include "llvm/Support/TargetSelect.h"