diff options
| author | kickbutt <kickbutt@yandex-team.com> | 2024-01-23 23:36:43 +0300 |
|---|---|---|
| committer | kickbutt <kickbutt@yandex-team.com> | 2024-01-23 23:55:22 +0300 |
| commit | fe742a0b69a530f86d1ea7aa84978d673256f8b7 (patch) | |
| tree | a045a5eb8dba770797e84d0b233098605396027d /build | |
| parent | bd7d89b121ae7b9f4427766292c950fcc91c2975 (diff) | |
| download | ydb-fe742a0b69a530f86d1ea7aa84978d673256f8b7.tar.gz | |
Fix separator in CUDA_ARCHITECTURES
Diffstat (limited to 'build')
| -rw-r--r-- | build/scripts/link_exe.py | 5 | ||||
| -rwxr-xr-x | build/ymake_conf.py | 4 |
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 |