Skip to content

Commit e915948

Browse files
committed
BUILD: Fix UCX GPU device API detection
Signed-off-by: Michal Shalev <[email protected]>
1 parent b21955c commit e915948

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

meson.build

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ endif
116116
if cuda_dep.found()
117117
nvcc_cmd = find_program('nvcc', required: false)
118118
if nvcc_cmd.found()
119-
if nvcc_cmd.version().version_compare('>=12.8') and nvcc_cmd.version().version_compare('<13.0')
119+
if nvcc.version().version_compare('>=12.8') and nvcc.version().version_compare('<13.0')
120120
doca_gpunetio_dep = dependency('doca-gpunetio', required : false)
121121
else
122122
warning('CUDA version = ' + nvcc_cmd.version() + ', GPUNETIO plugin will be disabled')
@@ -201,7 +201,7 @@ if ucx_dep.found() and cuda_dep.found() and nvcc_prog.found()
201201
have_gpu_side = cuda.compiles('''
202202
#include <ucp/api/device/ucp_device_impl.h>
203203
int main() { return 0; }
204-
''', dependencies : ucx_dep, args: nvcc_flags)
204+
''', dependencies : [ucx_dep, doca_gpunetio_dep], args: nvcc_flags)
205205

206206
have_host_side = cpp.compiles('''
207207
#include <ucp/api/device/ucp_host.h>
@@ -218,6 +218,7 @@ if ucx_dep.found() and cuda_dep.found() and nvcc_prog.found()
218218
'GPU-side compile' : have_gpu_side,
219219
'Host-side compile' : have_host_side,
220220
'nvcc available' : nvcc_prog.found(),
221+
'DOCA GPUNETIO found': doca_gpunetio_dep.found(),
221222
}, section: 'UCX GPU Device API', bool_yn: true)
222223
endif
223224

0 commit comments

Comments
 (0)