aboutsummaryrefslogtreecommitdiffstats
path: root/build
diff options
context:
space:
mode:
authorkickbutt <kickbutt@yandex-team.com>2024-01-23 23:36:43 +0300
committerAlexander Smirnov <alex@ydb.tech>2024-01-24 15:02:13 +0300
commit299dc21a6f70a16b00e1d564fa56961331552415 (patch)
tree183f498a2e98a04f9a6143a11006ba0c10f867f3 /build
parent5ccb6ae864be6b53df427dc3a0fa14dbb95aa11c (diff)
downloadydb-299dc21a6f70a16b00e1d564fa56961331552415.tar.gz
Fix separator in CUDA_ARCHITECTURES
Diffstat (limited to 'build')
-rw-r--r--build/scripts/link_exe.py5
-rwxr-xr-xbuild/ymake_conf.py4
2 files changed, 5 insertions, 4 deletions
diff --git a/build/scripts/link_exe.py b/build/scripts/link_exe.py
index a5744e5f25..980ac05d7e 100644
--- a/build/scripts/link_exe.py
+++ b/build/scripts/link_exe.py
@@ -55,7 +55,7 @@ def prune_cuda_libraries(cmd, prune_arches, nvprune_exe, build_root):
tmp_names_gen = name_generator('cuda_pruned_libs')
arch_args = []
- for arch in prune_arches.split(','):
+ for arch in prune_arches.split(':'):
arch_args.append('-gencode')
arch_args.append('arch={},code={}'.format(compute_arch(arch), arch))
@@ -200,7 +200,8 @@ def parse_args():
parser.add_option('--source-root')
parser.add_option('--clang-ver')
parser.add_option('--dynamic-cuda', action='store_true')
- parser.add_option('--cuda-architectures')
+ parser.add_option('--cuda-architectures',
+ help='List of supported CUDA architectures, separated by ":" (e.g. "sm_52:compute_70:lto_90a"')
parser.add_option('--nvprune-exe')
parser.add_option('--build-root')
parser.add_option('--arch')
diff --git a/build/ymake_conf.py b/build/ymake_conf.py
index acf6dfd9a0..33c55da3d5 100755
--- a/build/ymake_conf.py
+++ b/build/ymake_conf.py
@@ -2396,7 +2396,7 @@ class Cuda(object):
host, target = self.build.host_target
if not target.is_linux_x86_64:
# do not impose any restrictions, when build not for "linux 64-bit"
- return []
+ return ''
# do not include 'lto' type,
# because we already perform static linking
@@ -2422,7 +2422,7 @@ class Cuda(object):
for typ in supported_types
for ver in supported_vers]
- return ','.join(cuda_architectures)
+ return ':'.join(cuda_architectures)
def auto_use_arcadia_cuda(self):
return not self.cuda_root.from_user