Skip to content

Commit

Permalink
[Cherry-pick] Support nvcc lazy on for cpu. (#45090)
Browse files Browse the repository at this point in the history
* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__

* nvcc_lazy__
  • Loading branch information
xiaoxiaohehe001 authored Aug 12, 2022
1 parent 1bec83f commit 4596b9a
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cmake/experiments/cuda_module_loading_lazy.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down
1 change: 1 addition & 0 deletions tools/nvcc_lazy.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 4596b9a

Please sign in to comment.