From 4596b9a22540fb0ea5d369c3c804544de61d03d0 Mon Sep 17 00:00:00 2001
From: xiaoxiaohehe001 <49090790+xiaoxiaohehe001@users.noreply.github.com>
Date: Fri, 12 Aug 2022 17:48:44 +0800
Subject: [PATCH] [Cherry-pick] Support nvcc lazy on for cpu. (#45090)

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__
---
 cmake/experiments/cuda_module_loading_lazy.cmake | 4 ++--
 tools/nvcc_lazy.sh                               | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/cmake/experiments/cuda_module_loading_lazy.cmake b/cmake/experiments/cuda_module_loading_lazy.cmake
index 531efb52da..d1e07f57cb 100644
--- a/cmake/experiments/cuda_module_loading_lazy.cmake
+++ b/cmake/experiments/cuda_module_loading_lazy.cmake
@@ -23,8 +23,8 @@ if(LINUX)
     )
     return()
   endif()
-  if(NOT ${CUDA_FOUND})
-    message("EXP_CUDA_MODULE_LOADING_LAZY only works with CUDA")
+  if(NOT WITH_GPU)
+    message("EXP_CUDA_MODULE_LOADING_LAZY only works with GPU")
     return()
   endif()
   if(${CUDA_VERSION} VERSION_GREATER_EQUAL "11.7")
diff --git a/tools/nvcc_lazy.sh b/tools/nvcc_lazy.sh
index efb0223ae6..7795836f8c 100755
--- a/tools/nvcc_lazy.sh
+++ b/tools/nvcc_lazy.sh
@@ -36,6 +36,7 @@ echo -e >> $1
 echo "## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY" >> $1
 echo -e >> $1
 echo "# set cicc PATH for Centos" >> $1
+echo "export PATH=\$PATH:$2/bin" >> $1
 echo "export PATH=\$PATH:$2/nvvm/bin" >> $1
 echo -e >> $1
 echo "# check nvcc version, if nvcc >= 11.7, just run nvcc itself" >> $1
-- 
GitLab