let user specify CUDA_HOST_COMPILER

Summary: Pull Request resolved: pytorch#32904 Differential Revision: D19729047 Pulled By: ezyang fbshipit-source-id: c233e3924f71a025c51d25a7e3a8d728dac8730a
gaurav1086 · Feb 4, 2020 · 27e1fec · 27e1fec
1 parent d3a0bdd
commit 27e1fec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/setup_helpers/cmake.py b/tools/setup_helpers/cmake.py
@@ -227,9 +227,11 @@ def generate(self, version, cmake_python_library, build_python, build_test, my_e
             var: var for var in
             ('BLAS',
              'BUILDING_WITH_TORCH_LIBS',
+             'CUDA_HOST_COMPILER',
              'CUDA_NVCC_EXECUTABLE',
              'CUDNN_LIBRARY',
              'CUDNN_INCLUDE_DIR',
+             'CUDNN_ROOT',
              'EXPERIMENTAL_SINGLE_THREAD_POOL',
              'INSTALL_TEST',
              'JAVA_HOME',