diff options
| author | Jon Crall <erotemic@gmail.com> | 2018-01-11 07:24:17 -0500 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-01-11 07:24:17 -0500 |
| commit | 94f439c07c4f3e13fbd87016212b3534d31a7165 (patch) | |
| tree | 303e4ac743b1bfa48d4f6f5e18e7857f3b661e29 /torch | |
| parent | 0988e328c983b436a25cdcdd1fbbfcf4c054de43 (diff) | |
| download | pytorch-94f439c07c4f3e13fbd87016212b3534d31a7165.tar.gz pytorch-94f439c07c4f3e13fbd87016212b3534d31a7165.tar.bz2 pytorch-94f439c07c4f3e13fbd87016212b3534d31a7165.zip | |
Fixed setup.py to handle CUDNN_LIBRARY envvar with aten (#4597)
* Fixed setup.py to handle CUDNN_LIBRARY envvar with aten
* undo changes
* Added CUDNN_LIBRARY to bat file
Diffstat (limited to 'torch')
| -rw-r--r-- | torch/lib/build_libs.bat | 1 | ||||
| -rwxr-xr-x | torch/lib/build_libs.sh | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/torch/lib/build_libs.bat b/torch/lib/build_libs.bat index f51a081463..0a265c18ab 100644 --- a/torch/lib/build_libs.bat +++ b/torch/lib/build_libs.bat @@ -124,6 +124,7 @@ goto:eof -DNO_NNPACK=%NO_NNPACK% ^ -DCUDNN_INCLUDE_DIR="%CUDNN_INCLUDE_DIR%" ^ -DCUDNN_LIB_DIR="%CUDNN_LIB_DIR%" ^ + -DCUDNN_LIBRARY="%CUDNN_LIBRARY%" ^ -DCMAKE_BUILD_TYPE=%BUILD_TYPE% %MAKE_COMMAND% diff --git a/torch/lib/build_libs.sh b/torch/lib/build_libs.sh index 71e3bbc44f..5129e6dba3 100755 --- a/torch/lib/build_libs.sh +++ b/torch/lib/build_libs.sh @@ -161,6 +161,7 @@ function build_aten() { -DNO_NNPACK=$((1-$WITH_NNPACK)) \ -DCUDNN_INCLUDE_DIR=$CUDNN_INCLUDE_DIR \ -DCUDNN_LIB_DIR=$CUDNN_LIB_DIR \ + -DCUDNN_LIBRARY=$CUDNN_LIBRARY \ -DATEN_NO_CONTRIB=1 \ -DCMAKE_INSTALL_PREFIX="$INSTALL_DIR" \ -DCMAKE_EXPORT_COMPILE_COMMANDS=1 |