diff options
Diffstat (limited to 'torch/lib')
| -rwxr-xr-x | torch/lib/build_libs.sh | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/torch/lib/build_libs.sh b/torch/lib/build_libs.sh index 1948c3f9e2..dfa8c34285 100755 --- a/torch/lib/build_libs.sh +++ b/torch/lib/build_libs.sh @@ -123,6 +123,7 @@ function build_nccl() { mkdir -p build/nccl cd build/nccl ${CMAKE_VERSION} ../../nccl -DCMAKE_MODULE_PATH="$BASE_DIR/cmake/FindCUDA" \ + ${CMAKE_GENERATOR} \ -DCMAKE_BUILD_TYPE=Release \ -DCMAKE_INSTALL_PREFIX="$INSTALL_DIR" \ -DCMAKE_C_FLAGS="$C_FLAGS" \ |