Created
November 6, 2017 16:49
-
-
Save killeent/c78ddf12a6e45ab30ae4cc3efa5a4d2a to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
diff --git a/aten/CMakeLists.txt b/aten/CMakeLists.txt | |
index 9cccd34..136ce27 100644 | |
--- a/aten/CMakeLists.txt | |
+++ b/aten/CMakeLists.txt | |
@@ -70,5 +70,10 @@ include_directories( | |
${CMAKE_CURRENT_SOURCE_DIR}/src | |
${CMAKE_CURRENT_BINARY_DIR}/src/ATen) | |
add_subdirectory(src/ATen/test) | |
-add_subdirectory(contrib/data) | |
-add_subdirectory(contrib/meter) | |
+ | |
+if(ATEN_NO_CONTRIB) | |
+ message("disable contrib because ATEN_NO_CONTRIB is set") | |
+else() | |
+ add_subdirectory(contrib/data) | |
+ add_subdirectory(contrib/meter) | |
+endif() | |
diff --git a/torch/lib/build_libs.sh b/torch/lib/build_libs.sh | |
index e93415c..53d9a8b 100755 | |
--- a/torch/lib/build_libs.sh | |
+++ b/torch/lib/build_libs.sh | |
@@ -143,6 +143,7 @@ function build_aten() { | |
${CMAKE_VERSION} ../../../../aten \ | |
-DCMAKE_BUILD_TYPE=Release \ | |
-DNO_CUDA=$((1-$WITH_CUDA)) \ | |
+ -DATEN_NO_CONTRIB=1 \ | |
-DCMAKE_INSTALL_PREFIX="$INSTALL_DIR" | |
# purpusefully not passing C_FLAGS for the same reason as above | |
make -j$(getconf _NPROCESSORS_ONLN) install |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment