torch-mlir/lib/CAPI/CMakeLists.txt

39 lines
737 B
CMake

add_mlir_public_c_api_library(TorchMLIRCAPI
Dialects.cpp
Registration.cpp
TorchOps.cpp
TorchTypes.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/torch-mlir-c/
ENABLE_AGGREGATION
LINK_COMPONENTS
Core
LINK_LIBS PUBLIC
MLIRIR
MLIRAffineToStandard
MLIRArithmeticToLLVM
MLIRArithmeticTransforms
MLIRBufferizationTransforms
MLIRControlFlowToLLVM
MLIRFuncToLLVM
MLIRFuncTransforms
MLIRLinalgToLLVM
MLIRLinalgTransforms
MLIRMathToLLVM
MLIRMemRefToLLVM
MLIRReconcileUnrealizedCasts
MLIRSCFToControlFlow
MLIRSCFTransforms
MLIRTensorTransforms
MLIRTosaToArith
MLIRTosaToLinalg
MLIRSupport
TorchMLIRTorchDialect
TorchMLIRInitAll
)
torch_mlir_target_includes(TorchMLIRCAPI)