diff options
| author | Zachary DeVito <zdevito@gmail.com> | 2019-01-29 16:36:08 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-29 17:29:40 -0800 |
| commit | 21193bf12314f9d1be5ea36cbce3162c54b35225 (patch) | |
| tree | 417ef8696af531bb52706ac2c9bbb0c2581ebd4a /aten | |
| parent | ffed8bff6a1df334e371957009dd77af0c5140dd (diff) | |
| download | pytorch-21193bf12314f9d1be5ea36cbce3162c54b35225.tar.gz pytorch-21193bf12314f9d1be5ea36cbce3162c54b35225.tar.bz2 pytorch-21193bf12314f9d1be5ea36cbce3162c54b35225.zip | |
try to get rid of tmp_install (#16414)
Summary:
Rehash of previous attempts. This tries a different approach where we accept the install as specified in cmake (leaving bin/ include/ and lib/ alone), and then try to adjust the rest of the files to this more standard layout.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16414
Differential Revision: D13863635
Pulled By: zdevito
fbshipit-source-id: 23725f5c64d7509bf3ca8f472dcdcad074de9828
Diffstat (limited to 'aten')
| -rw-r--r-- | aten/CMakeLists.txt | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/aten/CMakeLists.txt b/aten/CMakeLists.txt index c479a0ca8b..88af4e5351 100644 --- a/aten/CMakeLists.txt +++ b/aten/CMakeLists.txt @@ -77,6 +77,11 @@ else() message("disabling CUDA because USE_CUDA is set false") SET(AT_CUDA_ENABLED 0) endif() +if(NOT USE_CUDA) + # we still parse THCUNN even if cuda is disabled to make sure to + # install it + INSTALL(FILES src/THCUNN/generic/THCUNN.h DESTINATION "${ATEN_INSTALL_INCLUDE_SUBDIR}/THCUNN/generic") +endif() if(NOT USE_NNPACK) set(AT_NNPACK_ENABLED 0) |