Skip to content

Commit cae30df

Browse files
committed
meson.build: add dependency on cublas and cusparse
1 parent 74725ea commit cae30df

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

meson.build

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -288,6 +288,16 @@ endif
288288
if get_option('use_gpu')
289289
gpu_c_args = common_c_args + ['-DPY_GPU=1', '-DINDIRECT=1']
290290
_deps += cuda_dep
291+
cublas_dep = cc.find_library('cublas', required: true)
292+
if not cublas_dep.found()
293+
cublas_dep = dependency('cublas', required: true)
294+
endif
295+
_deps += cublas_dep
296+
cusparse_dep = cc.find_library('cusparse', required: true)
297+
if not cusparse_dep.found()
298+
cusparse_dep = dependency('cusparse', required: true)
299+
endif
300+
_deps += cusparse_dep
291301
if get_option('gpu_atrans')
292302
gpu_c_args += '-DGPU_TRANSPOSE_MAT=1'
293303
endif

0 commit comments

Comments
 (0)