Unverified Commit 0726685f authored by Kasper Marstal's avatar Kasper Marstal Committed by GitHub
Browse files

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

parents 9bd78006 de7bd90c
......@@ -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 )
......
......@@ -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>
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment