未验证 提交 4596b9a2 编写于 作者: X xiaoxiaohehe001 提交者: GitHub

[Cherry-pick] Support nvcc lazy on for cpu. (#45090)

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__
上级 1bec83f4
...@@ -23,8 +23,8 @@ if(LINUX) ...@@ -23,8 +23,8 @@ if(LINUX)
) )
return() return()
endif() endif()
if(NOT ${CUDA_FOUND}) if(NOT WITH_GPU)
message("EXP_CUDA_MODULE_LOADING_LAZY only works with CUDA") message("EXP_CUDA_MODULE_LOADING_LAZY only works with GPU")
return() return()
endif() endif()
if(${CUDA_VERSION} VERSION_GREATER_EQUAL "11.7") if(${CUDA_VERSION} VERSION_GREATER_EQUAL "11.7")
......
...@@ -36,6 +36,7 @@ echo -e >> $1 ...@@ -36,6 +36,7 @@ echo -e >> $1
echo "## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY" >> $1 echo "## CUDA_MODULE_LOADING=EAGER,DEFAULT,LAZY" >> $1
echo -e >> $1 echo -e >> $1
echo "# set cicc PATH for Centos" >> $1 echo "# set cicc PATH for Centos" >> $1
echo "export PATH=\$PATH:$2/bin" >> $1
echo "export PATH=\$PATH:$2/nvvm/bin" >> $1 echo "export PATH=\$PATH:$2/nvvm/bin" >> $1
echo -e >> $1 echo -e >> $1
echo "# check nvcc version, if nvcc >= 11.7, just run nvcc itself" >> $1 echo "# check nvcc version, if nvcc >= 11.7, just run nvcc itself" >> $1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册