diff --git a/CMakeLists.txt b/CMakeLists.txt index 61d47e2ccc087dba9b02c28c0130965a23350a2d..a8d42599c84215d93246fbae5a6badb7e07f5d75 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -108,6 +108,7 @@ find_package( MPI QUIET ) find_package( Wt QUIET ) find_package( OpenMP QUIET ) find_package( KOKKOS QUIET ) +find_package( TORCH QUIET ) if( ${CMAKE_VERSION} VERSION_LESS "3.12.0" ) find_package( PythonInterp ) diff --git a/cmake/Modules/FindTORCH.cmake b/cmake/Modules/FindTORCH.cmake new file mode 100644 index 0000000000000000000000000000000000000000..4e1a9a74ce3f7417ce81f6a1e26b8fdcb5bbd3ec --- /dev/null +++ b/cmake/Modules/FindTORCH.cmake @@ -0,0 +1,29 @@ +# - Try to find the Torch library +# +# Once done this will define +# +# TORCH_FOUND - system has Torch +# TORCH_INCLUDE_DIR - the Torch include directory +# TORCH_LIBRARIES - the libraries needed to use Torch + +find_path( TORCH_INCLUDE_DIR torch.h + PATHS + /usr/include/torch + /usr/local/include/torch + /usr/include/torch/csrc/api/include/torch ) + +find_library( TORCH_LIBRARIES torch + PATHS + /usr/lib + /usr/local/lib ) + +# handle the QUIETLY and REQUIRED arguments and set KOKKOS_FOUND to TRUE if +# all listed variables are TRUE +include( FindPackageHandleStandardArgs ) +FIND_PACKAGE_HANDLE_STANDARD_ARGS( TORCH REQUIRED_VARS TORCH_INCLUDE_DIR TORCH_LIBRARIES ) + +if ( TORCH_FOUND ) +message( STATUS "Torch found" ) +endif( TORCH_FOUND ) + +mark_as_advanced( TORCH_LIBRARIES TORCH_INCLUDE_DIR )