BUILD: add nvcc_gencode option for CUDA architectures

Author: michal-shalev

meson.build

@@ -67,20 +67,20 @@ if cuda_dep.found()
     cuda = import('unstable-cuda')
     nvcc = meson.get_compiler('cuda')
 
-    # Please extend with your arch if not present in the list
     nvcc_flags = []
-    nvcc_flags += ['-gencode', 'arch=compute_80,code=sm_80']
-    nvcc_flags += ['-gencode', 'arch=compute_90,code=sm_90']
+    nvcc_flags_link = []
+    if get_option('nvcc_gencode') != ''
+      foreach p : get_option('nvcc_gencode').split(' ')
+        nvcc_flags += p
+        nvcc_flags_link += p
+      endforeach
+    endif
     add_project_arguments(nvcc_flags, language: 'cuda')
+    add_project_link_arguments(nvcc_flags_link, language: 'cuda')
 
     # Refer to https://mesonbuild.com/Cuda-module.html
     add_project_arguments('-forward-unknown-to-host-compiler', language: 'cuda')
     add_project_arguments('-rdc=true', language: 'cuda')
-
-    nvcc_flags_link = []
-    nvcc_flags_link += ['-gencode=arch=compute_80,code=sm_80']
-    nvcc_flags_link += ['-gencode=arch=compute_90,code=sm_90']
-    add_project_link_arguments(nvcc_flags_link, language: 'cuda')
 endif
 
 # DOCA

meson_options.txt

@@ -23,6 +23,8 @@ option('gds_path', type: 'string', value: '/usr/local/cuda/', description: 'Path
 option('cudapath_inc', type: 'string', value: '', description: 'Include path for CUDA')
 option('cudapath_lib', type: 'string', value: '', description: 'Library path for CUDA')
 option('cudapath_stub', type: 'string', value: '', description: 'Extra Stub path for CUDA')
+option('nvcc_gencode', type: 'string', value: '-gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_90,code=sm_90',
+       description: 'NVCC GPU architectures to target (space-separated -gencode flags)')
 option('static_plugins', type: 'string', value: '', description: 'Plugins to be built-in, comma-separated')
 option('build_docs', type: 'boolean', value: false, description: 'Build Doxygen documentation')
 option('log_level', type: 'combo', choices: ['trace', 'debug', 'info', 'warning', 'error', 'fatal', 'auto'], value: 'auto', description: 'Log Level (auto: auto-detect based on build type: trace for debug builds, info for release builds)')