From 92d14840e520954f1f7722afec2feac2d576f51b Mon Sep 17 00:00:00 2001
From: xiaoxiaohehe001 <49090790+xiaoxiaohehe001@users.noreply.github.com>
Date: Wed, 3 Aug 2022 11:08:22 +0800
Subject: [PATCH] nvcc_centos (#44832)

---
 tools/nvcc_lazy | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/nvcc_lazy b/tools/nvcc_lazy
index e3e7e36102..a553c6f5dc 100755
--- a/tools/nvcc_lazy
+++ b/tools/nvcc_lazy
@@ -16,9 +16,11 @@ set -e
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 ## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY
 
+# set cicc PATH for Centos
+export PATH=$PATH:/usr/local/cuda/nvvm/bin
+
 # check nvcc version, if nvcc >= 11.7, just run nvcc itself
 CUDA_VERSION=$(nvcc --version | grep -oP '(?<=V)\d*\.\d*')
 CUDA_VERSION_MAJOR=${CUDA_VERSION%.*}
-- 
GitLab