2022-03-03 00:48:15 +08:00
|
|
|
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
|
2022-03-03 00:48:15 +08:00
|
|
|
TorchMLIRTorchDialect
|
|
|
|
TorchMLIRTMTensorDialect
|
2023-04-12 02:18:49 +08:00
|
|
|
TorchMLIRTorchUtils
|
2022-03-03 00:48:15 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
torch_mlir_target_includes(TorchMLIRTorchToTMTensor)
|