mirror of https://github.com/llvm/torch-mlir
55 lines
1.1 KiB
CMake
55 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()
|
|
|
|
# include_directories(
|
|
# ${PYTHON_INCLUDE_DIRS}
|
|
# )
|
|
|
|
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
|
|
NPCOMPCAPI
|
|
|
|
# Core dialects
|
|
MLIRSCF
|
|
|
|
# Upstream depends
|
|
MLIRDialect
|
|
MLIREDSC
|
|
MLIRIR
|
|
MLIRLLVMIR
|
|
MLIRPass
|
|
MLIRTransforms
|
|
)
|
|
|
|
npcomp_python_target_compile_options(NPCOMPPythonCommon)
|