aboutsummaryrefslogtreecommitdiffstats
path: root/build/scripts/link_exe.py
diff options
context:
space:
mode:
authordeshevoy <deshevoy@yandex-team.com>2024-06-13 00:04:21 +0300
committerdeshevoy <deshevoy@yandex-team.com>2024-06-13 00:14:04 +0300
commit3069c8835032617846ef6fc3a911daf67b9d3f72 (patch)
tree50189447fd64410f4cd29dfdfdd2e81ea1d4c373 /build/scripts/link_exe.py
parenta138d1c8e540b93078ba6736790b57c94f8ae635 (diff)
downloadydb-3069c8835032617846ef6fc3a911daf67b9d3f72.tar.gz
[build] link_exe: Add cupti to CUDA_LIBRARIES
bddf00bb1bd387adf17502c2af8be1e265c8f5a8
Diffstat (limited to 'build/scripts/link_exe.py')
-rw-r--r--build/scripts/link_exe.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/build/scripts/link_exe.py b/build/scripts/link_exe.py
index 25fe116652..c1533c4dfc 100644
--- a/build/scripts/link_exe.py
+++ b/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