diff options
author | kickbutt <kickbutt@yandex-team.com> | 2024-01-23 23:36:43 +0300 |
---|---|---|
committer | Alexander Smirnov <alex@ydb.tech> | 2024-01-24 15:02:13 +0300 |
commit | 299dc21a6f70a16b00e1d564fa56961331552415 (patch) | |
tree | 183f498a2e98a04f9a6143a11006ba0c10f867f3 /build/scripts/link_exe.py | |
parent | 5ccb6ae864be6b53df427dc3a0fa14dbb95aa11c (diff) | |
download | ydb-299dc21a6f70a16b00e1d564fa56961331552415.tar.gz |
Fix separator in CUDA_ARCHITECTURES
Diffstat (limited to 'build/scripts/link_exe.py')
-rw-r--r-- | build/scripts/link_exe.py | 5 |
1 files changed, 3 insertions, 2 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') |