Browse Source

[build] link_exe: Add cupti to CUDA_LIBRARIES
bddf00bb1bd387adf17502c2af8be1e265c8f5a8

deshevoy 9 months ago
parent
commit
3069c88350
1 changed files with 2 additions and 0 deletions
  1. 2 0
      build/scripts/link_exe.py

+ 2 - 0
build/scripts/link_exe.py

@@ -32,6 +32,7 @@ CUDA_LIBRARIES = {
     '-lcudart_static': '-lcudart',
     '-lcudnn_static': '-lcudnn',
     '-lcufft_static_nocallback': '-lcufft',
+    '-lcupti_static': '-lcupti',
     '-lcurand_static': '-lcurand',
     '-lcusolver_static': '-lcusolver',
     '-lcusparse_static': '-lcusparse',
@@ -83,6 +84,7 @@ class CUDAManager:
     def _known_fatbin_libs(self, libs):
         libs_wo_device_code = {
             '-lcudart_static',
+            '-lcupti_static',
             '-lnppc_static',
         }
         return set(libs) - libs_wo_device_code