|
94 | 94 | "as specify by the --sysroot configuration option"), |
95 | 95 | ('mpi_cmd_prefix', "Prefix command for running MPI programs (with default number of ranks)"), |
96 | 96 | ('cuda_compute_capabilities', "Comma-separated list of CUDA compute capabilities, as specified via " |
97 | | - "--cuda-compute-capabilities configuration option or via cuda_compute_capabilities easyconfig parameter"), |
| 97 | + "--cuda-compute-capabilities configuration option or via cuda_cc easyconfig parameter"), |
98 | 98 | ('cuda_cc_cmake', "List of CUDA compute capabilities suitable for use with $CUDAARCHS in CMake 3.18+"), |
99 | 99 | ('cuda_cc_space_sep', "Space-separated list of CUDA compute capabilities"), |
| 100 | + ('cuda_cc_space_sep_no_period', |
| 101 | + "Space-separated list of CUDA compute capabilities, without periods (e.g. '80 90')."), |
100 | 102 | ('cuda_cc_semicolon_sep', "Semicolon-separated list of CUDA compute capabilities"), |
101 | 103 | ('cuda_sm_comma_sep', "Comma-separated list of sm_* values that correspond with CUDA compute capabilities"), |
102 | 104 | ('cuda_sm_space_sep', "Space-separated list of sm_* values that correspond with CUDA compute capabilities"), |
@@ -363,13 +365,14 @@ def template_constant_dict(config, ignore=None, skip_lower=None, toolchain=None) |
363 | 365 |
|
364 | 366 | # step 6. CUDA compute capabilities |
365 | 367 | # Use the commandline / easybuild config option if given, else use the value from the EC (as a default) |
366 | | - cuda_compute_capabilities = build_option('cuda_compute_capabilities') or config.get('cuda_compute_capabilities') |
367 | | - if cuda_compute_capabilities: |
368 | | - template_values['cuda_compute_capabilities'] = ','.join(cuda_compute_capabilities) |
369 | | - template_values['cuda_cc_space_sep'] = ' '.join(cuda_compute_capabilities) |
370 | | - template_values['cuda_cc_semicolon_sep'] = ';'.join(cuda_compute_capabilities) |
371 | | - template_values['cuda_cc_cmake'] = ';'.join(cc.replace('.', '') for cc in cuda_compute_capabilities) |
372 | | - sm_values = ['sm_' + cc.replace('.', '') for cc in cuda_compute_capabilities] |
| 368 | + cuda_cc = build_option('cuda_compute_capabilities') or config.get('cuda_compute_capabilities') |
| 369 | + if cuda_cc: |
| 370 | + template_values['cuda_compute_capabilities'] = ','.join(cuda_cc) |
| 371 | + template_values['cuda_cc_space_sep'] = ' '.join(cuda_cc) |
| 372 | + template_values['cuda_cc_space_sep_no_period'] = ' '.join(cc.replace('.', '') for cc in cuda_cc) |
| 373 | + template_values['cuda_cc_semicolon_sep'] = ';'.join(cuda_cc) |
| 374 | + template_values['cuda_cc_cmake'] = ';'.join(cc.replace('.', '') for cc in cuda_cc) |
| 375 | + sm_values = ['sm_' + cc.replace('.', '') for cc in cuda_cc] |
373 | 376 | template_values['cuda_sm_comma_sep'] = ','.join(sm_values) |
374 | 377 | template_values['cuda_sm_space_sep'] = ' '.join(sm_values) |
375 | 378 |
|
|
0 commit comments