Просмотр исходного кода

[build] link_exe: Add cupti to CUDA_LIBRARIES
bddf00bb1bd387adf17502c2af8be1e265c8f5a8

deshevoy 9 месяцев назад
Родитель
Сommit
3069c88350
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      build/scripts/link_exe.py

+ 2 - 0
build/scripts/link_exe.py

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