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

19 lines
373 B
CMake

add_mlir_conversion_library(TorchMLIRTorchToTensor
TorchToTensor.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToTensor
DEPENDS
TorchMLIRConversionPassIncGen
LINK_LIBS PUBLIC
MLIRIR
MLIRPass
MLIRTensorDialect
TorchMLIRTorchDialect
TorchMLIRConversionUtils
)
torch_mlir_target_includes(TorchMLIRTorchToTensor)