diff options
| author | Zach DeVito <zdevito@fb.com> | 2017-12-04 12:16:37 -0800 |
|---|---|---|
| committer | Zachary DeVito <zdevito@gmail.com> | 2017-12-04 18:27:34 -0500 |
| commit | 7bf6aaacd379a3075d88b4402a086eeb5bc8d3a5 (patch) | |
| tree | 8395c773acd56efbfa80a692a18b7eac1cd9820e /torch/lib | |
| parent | d76f7a806df8451e3ed6bdca032fa743ce702e1d (diff) | |
| download | pytorch-7bf6aaacd379a3075d88b4402a086eeb5bc8d3a5.tar.gz pytorch-7bf6aaacd379a3075d88b4402a086eeb5bc8d3a5.tar.bz2 pytorch-7bf6aaacd379a3075d88b4402a086eeb5bc8d3a5.zip | |
add missing CMAKE_GENERATOR
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" \ |