Skip to content

Instantly share code, notes, and snippets.

@killeent
Created November 6, 2017 16:49
Show Gist options
  • Save killeent/c78ddf12a6e45ab30ae4cc3efa5a4d2a to your computer and use it in GitHub Desktop.
Save killeent/c78ddf12a6e45ab30ae4cc3efa5a4d2a to your computer and use it in GitHub Desktop.
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