| commit | 7251d70c5be1bd25f3fb9b2d1a1366541872a3fa | [log] [tgz] |
|---|---|---|
| author | Soumith Chintala <soumith@gmail.com> | Fri Jun 15 09:09:23 2018 -0400 |
| committer | Edward Z. Yang <ezyang@mit.edu> | Fri Jun 15 09:09:23 2018 -0400 |
| tree | 9ba3522e250d8e164b633dd64eeb79515e1b0c57 | |
| parent | 0965e8e9e721904094f9f1cc3e1b933ed625c345 [diff] |
fixed THD NO_CUDA (#8539)
diff --git a/torch/lib/THD/CMakeLists.txt b/torch/lib/THD/CMakeLists.txt index f9826b1..7c2a20e 100644 --- a/torch/lib/THD/CMakeLists.txt +++ b/torch/lib/THD/CMakeLists.txt
@@ -40,7 +40,6 @@ "XCode 8 or newer.") ENDIF(NOT HAS_THREAD_LOCAL) -FIND_PACKAGE(NCCL) FIND_PACKAGE(MPI) @@ -59,6 +58,8 @@ ADD_DEFINITIONS(-DWITH_CUDA=1) ENDIF() + + FIND_PACKAGE(NCCL) ENDIF() IF(MPI_FOUND)