diff --git a/tools/nvcc_lazy b/tools/nvcc_lazy index e3e7e361021c2f9ef81fc30d9d6875c9159e2a26..a553c6f5dcdfd02ea2fb0a8827ade4dac920da30 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%.*}