mirror of https://github.com/llvm/torch-mlir
19 lines
373 B
CMake
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)
|