mirror of https://github.com/llvm/torch-mlir
82 lines
3.0 KiB
CMake
82 lines
3.0 KiB
CMake
include(AddMLIRPython)
|
|
|
|
if(TORCH_MLIR_ENABLE_ONNX_C_IMPORTER)
|
|
add_subdirectory(onnx_c_importer)
|
|
endif()
|
|
|
|
################################################################################
|
|
# PyTorch
|
|
# Configure PyTorch if we have any features enabled which require it.
|
|
################################################################################
|
|
if(TORCH_MLIR_ENABLE_JIT_IR_IMPORTER OR TORCH_MLIR_ENABLE_LTC)
|
|
|
|
if (NOT TORCH_MLIR_USE_INSTALLED_PYTORCH)
|
|
# Source builds
|
|
message(STATUS "Building libtorch from source (features depend on it and NOT TORCH_MLIR_USE_INSTALLED_PYTORCH)")
|
|
set(ENV{TORCH_MLIR_SRC_PYTORCH_REPO} ${TORCH_MLIR_SRC_PYTORCH_REPO})
|
|
set(ENV{TORCH_MLIR_SRC_PYTORCH_BRANCH} ${TORCH_MLIR_SRC_PYTORCH_BRANCH})
|
|
set(ENV{TM_PYTORCH_INSTALL_WITHOUT_REBUILD} ${TM_PYTORCH_INSTALL_WITHOUT_REBUILD})
|
|
set(ENV{MACOSX_DEPLOYMENT_TARGET} ${MACOSX_DEPLOYMENT_TARGET})
|
|
set(ENV{CMAKE_OSX_ARCHITECTURES} ${CMAKE_OSX_ARCHITECTURES})
|
|
set(ENV{CMAKE_C_COMPILER_LAUNCHER} ${CMAKE_C_COMPILER_LAUNCHER})
|
|
set(ENV{CMAKE_CXX_COMPILER_LAUNCHER} ${CMAKE_CXX_COMPILER_LAUNCHER})
|
|
execute_process(
|
|
COMMAND ${TORCH_MLIR_SOURCE_DIR}/build_tools/build_libtorch.sh
|
|
RESULT_VARIABLE _result
|
|
)
|
|
if(_result)
|
|
message(FATAL_ERROR "Failed to run `build_libtorch.sh`")
|
|
endif()
|
|
set(TORCH_INSTALL_PREFIX "libtorch")
|
|
endif()
|
|
|
|
message(STATUS "Enabling PyTorch C++ dep (features depend on it)")
|
|
include(TorchMLIRPyTorch)
|
|
|
|
TorchMLIRProbeForPyTorchInstall()
|
|
if(TORCH_MLIR_USE_INSTALLED_PYTORCH)
|
|
TorchMLIRConfigurePyTorch()
|
|
else()
|
|
# Assume it is a sibling to the overall project.
|
|
set(Torch_DIR "${PROJECT_SOURCE_DIR}/../libtorch/share/cmake/Torch")
|
|
message(STATUS "Attempting to locate libtorch as a sibling to the project: ${Torch_DIR}")
|
|
if(NOT EXISTS "${Torch_DIR}")
|
|
message(FATAL_ERROR "Without TORCH_MLIR_USE_INSTALLED_PYTORCH, expected to find Torch configuration at ${Torch_DIR}, which does not exist")
|
|
endif()
|
|
endif()
|
|
|
|
find_package(Torch 1.11 REQUIRED)
|
|
|
|
set(TORCHGEN_DIR ${Torch_ROOT}/../../../torchgen)
|
|
|
|
include_directories(BEFORE
|
|
${TORCH_INCLUDE_DIRS}
|
|
${Python3_INCLUDE_DIRS}
|
|
)
|
|
link_directories("${TORCH_INSTALL_PREFIX}/lib")
|
|
message(STATUS "TORCH_CXXFLAGS is = ${TORCH_CXXFLAGS}")
|
|
if(${CMAKE_SYSTEM_NAME} STREQUAL "Linux" AND NOT TORCH_CXXFLAGS)
|
|
message(WARNING
|
|
"When building on Linux TORCH_CXXFLAGS are almost always required but were not detected. "
|
|
"It is very likely this this will produce a non-functional installation. "
|
|
"See notes in build_tools/cmake/TorchMLIRPyTorch.cmake.")
|
|
endif()
|
|
message(STATUS "TORCH_LIBRARIES = ${TORCH_LIBRARIES}")
|
|
endif()
|
|
|
|
# Include jit_ir_common if the jit_ir importer or LTC is enabled,
|
|
# since they both require it.
|
|
if(TORCH_MLIR_ENABLE_JIT_IR_IMPORTER OR TORCH_MLIR_ENABLE_LTC)
|
|
add_subdirectory(jit_ir_common)
|
|
endif()
|
|
|
|
# Include LTC.
|
|
if(TORCH_MLIR_ENABLE_LTC)
|
|
add_subdirectory(ltc)
|
|
endif()
|
|
|
|
# Include overall PT1 project.
|
|
if(TORCH_MLIR_ENABLE_PYTORCH_EXTENSIONS)
|
|
add_subdirectory(pt1)
|
|
endif()
|