From 65a95bc1194c55fb68139653e038bdd082c7d5f6 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Mon, 3 May 2021 00:01:15 -0400 Subject: [PATCH] [NVCC] Bugfix nvcc command tool that relies on the compile time env (#7964) * [NVCC] Bugfix nvcc command tool that relies on the compile time env * Update python/tvm/contrib/nvcc.py Co-authored-by: Cody Yu Co-authored-by: Cody Yu --- python/tvm/contrib/nvcc.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/python/tvm/contrib/nvcc.py b/python/tvm/contrib/nvcc.py index 99844f799d7a..6a7c09860b07 100644 --- a/python/tvm/contrib/nvcc.py +++ b/python/tvm/contrib/nvcc.py @@ -89,11 +89,13 @@ def compile_cuda(code, target="ptx", arch=None, options=None, path_target=None): cmd += ["-o", file_target] cmd += [temp_code] - cxx_compiler_path = tvm.support.libinfo().get("TVM_CXX_COMPILER_PATH") - if cxx_compiler_path != "": - # This tells nvcc where to find the c++ compiler just in case it is not in the path. - # On Windows it is not in the path by default. - cmd += ["-ccbin", cxx_compiler_path] + # NOTE: ccbin option can be used to tell nvcc where to find the c++ compiler + # just in case it is not in the path. On Windows it is not in the path by default. + # However, we cannot use TVM_CXX_COMPILER_PATH because the runtime env. + # Because it is hard to do runtime compiler detection, we require nvcc is configured + # correctly by default. + # if cxx_compiler_path != "": + # cmd += ["-ccbin", cxx_compiler_path] proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)