torch-mlir/lib/Conversion/TorchToTMTensor/CMakeLists.txt

21 lines
419 B
CMake
Raw Normal View History

add_mlir_conversion_library(TorchMLIRTorchToTMTensor
TorchToTMTensor.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToTMTensor
DEPENDS
TorchMLIRConversionPassIncGen
LINK_LIBS PUBLIC
MLIRIR
MLIRPass
2022-06-23 11:23:46 +08:00
MLIRLinalgDialect
MLIRMathDialect
TorchMLIRTorchDialect
TorchMLIRTMTensorDialect
TorchMLIRTorchUtils
)
torch_mlir_target_includes(TorchMLIRTorchToTMTensor)