Skip to content

Commit 594a8b6

Browse files
authored
CUSOLVER: Fix workspace size passing. (#1890)
The APIs expect a number of elements, not a number of bytes.
1 parent 7eda21f commit 594a8b6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/cusolver/dense.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ for (bname, fname, elty, relty) in ((:cusolverDnSgesvd_bufferSize, :cusolverDnSg
410410
devinfo = CuArray{Cint}(undef, 1)
411411
with_workspace($elty, bufferSize) do work
412412
$fname(dense_handle(), jobu, jobvt, m, n, A, lda, S, U, ldu, Vt, ldvt,
413-
work, sizeof(work), rwork, devinfo)
413+
work, length(work), rwork, devinfo)
414414
end
415415
unsafe_free!(rwork)
416416

@@ -469,7 +469,7 @@ for (bname, fname, elty, relty) in ((:cusolverDnSgesvdj_bufferSize, :cusolverDnS
469469
devinfo = CuArray{Cint}(undef, 1)
470470
with_workspace($elty, bufferSize) do work
471471
$fname(dense_handle(), jobz, econ, m, n, A, lda, S, U, ldu, V, ldv,
472-
work, sizeof(work), devinfo, params[])
472+
work, length(work), devinfo, params[])
473473
end
474474

475475
info = @allowscalar devinfo[1]

0 commit comments

Comments
 (0)