mirror of https://github.com/llvm/torch-mlir
51 lines
1.1 KiB
CMake
51 lines
1.1 KiB
CMake
################################################################################
|
|
# NPCOMPPythonCommon
|
|
################################################################################
|
|
|
|
include(AddLLVM)
|
|
include(NpcompPython)
|
|
|
|
# TODO: This should not be wired in at such a low/unconditional level.
|
|
# It is done here to be kept with the other LLVM initialization until a better
|
|
# place can be found for it.
|
|
set(ExtraInit_LIBADD)
|
|
if(NPCOMP_ENABLE_REFJIT)
|
|
llvm_map_components_to_libnames(refjit_llvm_libs
|
|
nativecodegen
|
|
)
|
|
message(STATUS "Including LLVM libs for RefJit: ${refjit_llvm_libs}")
|
|
list(APPEND ExtraInit_LIBADD
|
|
${refjit_llvm_libs})
|
|
endif()
|
|
|
|
set(PYBIND_SOURCES
|
|
MlirInit.cpp
|
|
MlirIr.cpp
|
|
MlirPass.cpp
|
|
NpcompDialect.cpp
|
|
CoreDialects.cpp
|
|
)
|
|
|
|
add_library(NPCOMPPythonCommon
|
|
${PYBIND_SOURCES}
|
|
)
|
|
|
|
target_link_libraries(NPCOMPPythonCommon
|
|
pybind11::module
|
|
|
|
NPCOMPInitAll
|
|
|
|
# Core dialects
|
|
MLIRSCF
|
|
|
|
# Upstream depends
|
|
MLIRDialect
|
|
MLIREDSC
|
|
MLIRIR
|
|
MLIRLLVMIR
|
|
MLIRPass
|
|
MLIRTransforms
|
|
)
|
|
|
|
npcomp_python_target_compile_options(NPCOMPPythonCommon)
|