diff options
| author | Bram Wasti <bwasti@fb.com> | 2017-01-05 09:52:45 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-01-05 09:52:45 -0800 |
| commit | 10bad040c27a0e22c7806c09289e89bda335ce73 (patch) | |
| tree | 0fa86ba60e0f4fbd1262e061aa19229671ccc329 /caffe2/contrib/nccl | |
| parent | ac03e6592901499be1b232a9f142275931fb095e (diff) | |
| parent | b31708fb6e5a3247d122b21c08f358bfe3ab3142 (diff) | |
| download | pytorch-10bad040c27a0e22c7806c09289e89bda335ce73.tar.gz pytorch-10bad040c27a0e22c7806c09289e89bda335ce73.tar.bz2 pytorch-10bad040c27a0e22c7806c09289e89bda335ce73.zip | |
Merge branch 'master' into cmake
Diffstat (limited to 'caffe2/contrib/nccl')
| -rw-r--r-- | caffe2/contrib/nccl/CMakeLists.txt | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/contrib/nccl/CMakeLists.txt b/caffe2/contrib/nccl/CMakeLists.txt index 0631cd12cd..97b3de6885 100644 --- a/caffe2/contrib/nccl/CMakeLists.txt +++ b/caffe2/contrib/nccl/CMakeLists.txt @@ -1,4 +1,4 @@ -if(HAVE_CUDA) +if(USE_CUDA) message(STATUS "Include NCCL operators") set(Caffe2_CONTRIB_NCCL_GPU_SRC "${CMAKE_CURRENT_SOURCE_DIR}/cuda_nccl_gpu.cc" |