Merge pull request #145 from SuperElastix/SELX-138-stop-ATen-build-if-python-is-not-found

SuperBuild/CMakeLists.txt

@@ -186,15 +186,8 @@ endif()
 # ---------------------------------------------------------------------
 # Build ATen
 
-find_package( PythonInterp )
-if( PYTHONINTERP_FOUND )
-  set( SUPERELASTIX_BUILD_ATEN_DEFAULT ON )
-else()
-  set( SUPERELASTIX_BUILD_ATEN_DEFAULT OFF )
-endif()
-
 mark_as_advanced( SUPERELASTIX_BUILD_ATEN )
-option( SUPERELASTIX_BUILD_ATEN "Build ATen." ${SUPERELASTIX_BUILD_ATEN_DEFAULT} )
+option( SUPERELASTIX_BUILD_ATEN "Build ATen." OFF )
 
 if( SUPERELASTIX_BUILD_ATEN )
   mark_as_advanced( USE_SYSTEM_ATEN )

SuperBuild/ExternalATen.cmake

@@ -18,14 +18,15 @@
 #=========================================================================
 
 set( proj ATen )
-set( ATen_REPOSITORY https://github.com/zdevito/ATen )
-set( ATen_TAG 635e9ef1f881391539e7426899370ebfe094fabd )
+set( ATen_REPOSITORY https://github.com/pytorch/pytorch )
+set( ATen_TAG 82e995e0b9bc73cedf98016b910297b44563342f )
 
 ExternalProject_Add( ${proj}
   GIT_REPOSITORY ${ATen_REPOSITORY}
   GIT_TAG ${ATen_TAG}
   UPDATE_COMMAND ""
   SOURCE_DIR ${proj}
+  SOURCE_SUBDIR aten
   BINARY_DIR ${proj}-build
   CMAKE_ARGS
     --no-warn-unused-cli
@@ -34,7 +35,7 @@ ExternalProject_Add( ${proj}
     -DBUILD_TESTING:BOOL=OFF
     -DCMAKE_BUILD_TYPE:STRING=${CMAKE_BUILD_TYPE}
     -DCMAKE_CONFIGURATION_TYPES:STRING=${CMAKE_CONFIGURATION_TYPES}
-    -DNO_CUDA=true
+    -DNO_CUDA:BOOL=TRUE
     -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
 )