mirror of https://github.com/llvm/torch-mlir
39 lines
998 B
CMake
39 lines
998 B
CMake
add_subdirectory(TorchToLinalg)
|
|
add_subdirectory(TorchToSCF)
|
|
add_subdirectory(TorchToArith)
|
|
add_subdirectory(TorchToMLProgram)
|
|
add_subdirectory(TorchToTosa)
|
|
if(TORCH_MLIR_ENABLE_MHLO)
|
|
add_subdirectory(TorchToMhlo)
|
|
endif()
|
|
add_subdirectory(TorchToTMTensor)
|
|
add_subdirectory(TorchConversionToMLProgram)
|
|
add_subdirectory(Utils)
|
|
|
|
# TODO: Automate this with add_torch_mlir_conversion_library.
|
|
set(linked_libs TorchMLIRTorchToLinalg
|
|
TorchMLIRTorchToSCF
|
|
TorchMLIRTorchToArith
|
|
TorchMLIRTorchToTosa
|
|
TorchMLIRTorchToTMTensor
|
|
TorchMLIRTorchToMLProgram
|
|
TorchMLIRTorchConversionToMLProgram
|
|
TorchMLIRConversionUtils)
|
|
if(TORCH_MLIR_ENABLE_MHLO)
|
|
list(APPEND linked_libs TorchMLIRTorchToMhlo)
|
|
endif()
|
|
|
|
add_mlir_library(TorchMLIRConversionPasses
|
|
Passes.cpp
|
|
|
|
DEPENDS
|
|
TorchMLIRConversionPassIncGen
|
|
|
|
LINK_COMPONENTS
|
|
Core
|
|
|
|
LINK_LIBS PUBLIC
|
|
${linked_libs}
|
|
#${torch_mlir_conversion_libs}
|
|
)
|