# FindTorch # ------- # # Finds the Torch library # # This will define the following variables: # # TORCH_FOUND -- True if the system has the Torch library # TORCH_INCLUDE_DIRS -- The include directories for torch # TORCH_LIBRARIES -- Libraries to link against # TORCH_CXX_FLAGS -- Additional (required) compiler flags # # and the following imported targets: # # torch macro(append_torchlib_if_found) foreach (_arg ${ARGN}) find_library(${_arg}_LIBRARY ${_arg} PATHS "${TORCH_INSTALL_PREFIX}/lib") if(${_arg}_LIBRARY) list(APPEND TORCH_LIBRARIES ${${_arg}_LIBRARY}) else() message(WARNING "static library ${${_arg}_LIBRARY} not found.") endif() endforeach() endmacro() macro(append_wholearchive_lib_if_found) foreach (_arg ${ARGN}) find_library(${_arg}_LIBRARY ${_arg} PATHS "${TORCH_INSTALL_PREFIX}/lib") if(${_arg}_LIBRARY) if("${CMAKE_CXX_COMPILER_ID}" MATCHES "Clang") list(APPEND TORCH_LIBRARIES "-Wl,-force_load,${${_arg}_LIBRARY}") elseif(MSVC) list(APPEND TORCH_LIBRARIES "-WHOLEARCHIVE:${${_arg}_LIBRARY}") else() # gcc list(APPEND TORCH_LIBRARIES "-Wl,--whole-archive ${${_arg}_LIBRARY} -Wl,--no-whole-archive") endif() else() message(WARNING "static library ${${_arg}_LIBRARY} not found.") endif() endforeach() endmacro() include(FindPackageHandleStandardArgs) if(DEFINED ENV{TORCH_INSTALL_PREFIX}) set(TORCH_INSTALL_PREFIX $ENV{TORCH_INSTALL_PREFIX}) else() # Assume we are in /share/cmake/Torch/TorchConfig.cmake get_filename_component(CMAKE_CURRENT_LIST_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH) get_filename_component(TORCH_INSTALL_PREFIX "${CMAKE_CURRENT_LIST_DIR}/../../../" ABSOLUTE) endif() # Include directories. if(EXISTS "${TORCH_INSTALL_PREFIX}/include") set(TORCH_INCLUDE_DIRS ${TORCH_INSTALL_PREFIX}/include ${TORCH_INSTALL_PREFIX}/include/torch/csrc/api/include) else() set(TORCH_INCLUDE_DIRS ${TORCH_INSTALL_PREFIX}/include ${TORCH_INSTALL_PREFIX}/include/torch/csrc/api/include) endif() # Library dependencies. if(@BUILD_SHARED_LIBS@) find_package(Caffe2 REQUIRED PATHS ${CMAKE_CURRENT_LIST_DIR}/../Caffe2) set(TORCH_LIBRARIES torch ${Caffe2_MAIN_LIBS}) append_torchlib_if_found(c10) else() add_library(torch STATIC IMPORTED) # set imported_location at the bottom #library need whole archive append_wholearchive_lib_if_found(torch torch_cpu) if(@USE_CUDA@) append_wholearchive_lib_if_found(torch_cuda c10_cuda) endif() # We need manually add dependent libraries when they are not linked into the # shared library. # TODO: this list might be incomplete. append_torchlib_if_found(c10) if(@BUILD_CAFFE2@) append_torchlib_if_found(Caffe2_perfkernels_avx512 Caffe2_perfkernels_avx2 Caffe2_perfkernels_avx) endif() if(@USE_NNPACK@) append_torchlib_if_found(nnpack) endif() if(@USE_PYTORCH_QNNPACK@) append_torchlib_if_found(pytorch_qnnpack) endif() if(@USE_QNNPACK@) append_torchlib_if_found(qnnpack) endif() if(@USE_XNNPACK@) append_torchlib_if_found(XNNPACK) endif() append_torchlib_if_found(caffe2_protos protobuf-lite protobuf protoc) append_torchlib_if_found(onnx onnx_proto) append_torchlib_if_found(foxi_loader fmt) append_torchlib_if_found(clog cpuinfo) if(NOT @USE_INTERNAL_PTHREADPOOL_IMPL@) append_torchlib_if_found(pthreadpool) endif() append_torchlib_if_found(eigen_blas) if(@USE_FBGEMM@) append_torchlib_if_found(fbgemm) endif() if(@USE_MKLDNN@) append_torchlib_if_found(dnnl mkldnn) endif() append_torchlib_if_found(sleef asmjit) endif() if(@USE_KINETO@) append_torchlib_if_found(kineto) endif() if(@USE_DEPLOY@) append_torchlib_if_found(torch_deploy) endif() if(@USE_CUDA@) if(MSVC) if(NOT NVTOOLEXT_HOME) set(NVTOOLEXT_HOME "C:/Program Files/NVIDIA Corporation/NvToolsExt") endif() if(DEFINED ENV{NVTOOLSEXT_PATH}) set(NVTOOLEXT_HOME $ENV{NVTOOLSEXT_PATH}) endif() set(TORCH_CUDA_LIBRARIES ${NVTOOLEXT_HOME}/lib/x64/nvToolsExt64_1.lib ${CUDA_LIBRARIES}) list(APPEND TORCH_INCLUDE_DIRS ${NVTOOLEXT_HOME}/include) find_library(CAFFE2_NVRTC_LIBRARY caffe2_nvrtc PATHS "${TORCH_INSTALL_PREFIX}/lib") list(APPEND TORCH_CUDA_LIBRARIES ${CAFFE2_NVRTC_LIBRARY}) elseif(APPLE) set(TORCH_CUDA_LIBRARIES ${CUDA_TOOLKIT_ROOT_DIR}/lib/libcudart.dylib ${CUDA_TOOLKIT_ROOT_DIR}/lib/libnvrtc.dylib ${CUDA_TOOLKIT_ROOT_DIR}/lib/libnvToolsExt.dylib ${CUDA_LIBRARIES}) else() find_library(LIBNVTOOLSEXT libnvToolsExt.so PATHS ${CUDA_TOOLKIT_ROOT_DIR}/lib64/) set(TORCH_CUDA_LIBRARIES ${CUDA_CUDA_LIB} ${CUDA_NVRTC_LIB} ${LIBNVTOOLSEXT} ${CUDA_LIBRARIES}) endif() if(@BUILD_SHARED_LIBS@) find_library(C10_CUDA_LIBRARY c10_cuda PATHS "${TORCH_INSTALL_PREFIX}/lib") list(APPEND TORCH_CUDA_LIBRARIES ${C10_CUDA_LIBRARY}) endif() list(APPEND TORCH_LIBRARIES ${TORCH_CUDA_LIBRARIES}) endif() # When we build libtorch with the old libstdc++ ABI, dependent libraries must too. if(CMAKE_SYSTEM_NAME STREQUAL "Linux") set(TORCH_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=@GLIBCXX_USE_CXX11_ABI@") endif() find_library(TORCH_LIBRARY torch PATHS "${TORCH_INSTALL_PREFIX}/lib") # the statements below changes target properties on # - the imported target from Caffe2Targets.cmake in shared library mode (see the find_package above) # - this is untested whether it is the correct (or desired) methodology in CMake # - the imported target created in this file in static library mode if(NOT @BUILD_SHARED_LIBS@) # do not set this property on the shared library target, as it will cause confusion in some builds # as the configuration specific property is set in the Caffe2Targets.cmake file set_target_properties(torch PROPERTIES IMPORTED_LOCATION "${TORCH_LIBRARY}" ) endif() set_target_properties(torch PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${TORCH_INCLUDE_DIRS}" CXX_STANDARD 14 ) if(TORCH_CXX_FLAGS) set_property(TARGET torch PROPERTY INTERFACE_COMPILE_OPTIONS "${TORCH_CXX_FLAGS}") endif() find_package_handle_standard_args(Torch DEFAULT_MSG TORCH_LIBRARY TORCH_INCLUDE_DIRS)