mirror of https://github.com/llvm/torch-mlir
68 lines
1.7 KiB
CMake
68 lines
1.7 KiB
CMake
add_subdirectory(CAPI)
|
|
add_subdirectory(Conversion)
|
|
add_subdirectory(Dialect)
|
|
add_subdirectory(RefBackend)
|
|
add_subdirectory(Python)
|
|
add_subdirectory(Typing)
|
|
|
|
if(NPCOMP_ENABLE_REFJIT)
|
|
add_subdirectory(Backend/RefJIT)
|
|
endif()
|
|
if(NPCOMP_ENABLE_IREE)
|
|
add_subdirectory(Backend/IREE)
|
|
endif()
|
|
|
|
################################################################################
|
|
# Setup the initialization target.
|
|
# This includes conditional dependencies based on whether features are enabled.
|
|
################################################################################
|
|
|
|
set(ALL_DEPENDS)
|
|
if(NPCOMP_ENABLE_IREE)
|
|
list(APPEND ALL_DEPENDS
|
|
iree_tools_init_compiler_modules
|
|
iree_tools_init_iree_passes_and_dialects
|
|
iree_tools_init_mlir_passes_and_dialects
|
|
iree_tools_init_targets
|
|
iree_tools_init_xla_dialects
|
|
)
|
|
endif()
|
|
|
|
get_property(mlir_dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
|
|
get_property(mlir_conversion_libs GLOBAL PROPERTY MLIR_CONVERSION_LIBS)
|
|
|
|
get_property(npcomp_dialect_libs GLOBAL PROPERTY NPCOMP_DIALECT_LIBS)
|
|
get_property(npcomp_conversion_libs GLOBAL PROPERTY NPCOMP_CONVERSION_LIBS)
|
|
|
|
message(STATUS "NPCOMP Dialect libs: ${npcomp_dialect_libs}")
|
|
message(STATUS "NPCOMP Conversion libs: ${npcomp_conversion_libs}")
|
|
|
|
add_npcomp_library(NPCOMPInitAll
|
|
InitAll.cpp
|
|
|
|
LINK_LIBS
|
|
|
|
PUBLIC
|
|
# Local depends
|
|
NPCOMPRefBackend
|
|
NPCOMPRefbackDialect
|
|
NPCOMPTCPDialect
|
|
NPCOMPTCFDialect
|
|
NPCOMPTorchDialect
|
|
NPCOMPRefbackrtDialect
|
|
NPCOMPATenDialect
|
|
NPCOMPBasicpyDialect
|
|
NPCOMPBasicpyPasses
|
|
NPCOMPConversionPasses
|
|
NPCOMPNumpyDialect
|
|
NPCOMPNumpyPasses
|
|
NPCOMPTCFPasses
|
|
NPCOMPTypingPasses
|
|
|
|
${npcomp_dialect_libs}
|
|
${npcomp_conversion_libs}
|
|
${mlir_dialect_libs}
|
|
${mlir_conversion_libs}
|
|
${ALL_DEPENDS}
|
|
)
|