Skip to content

Commit f62af73

Browse files
kshyattmaleadt
andauthored
Switch CUBLAS to device-side pointer mode (#2616)
Co-authored-by: Tim Besard <[email protected]>
1 parent 4d85f27 commit f62af73

File tree

12 files changed

+1319
-1180
lines changed

12 files changed

+1319
-1180
lines changed

lib/cublas/CUBLAS.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ function handle()
104104
end
105105

106106
cublasSetStream_v2(new_handle, cuda.stream)
107+
cublasSetPointerMode_v2(new_handle, CUBLAS_POINTER_MODE_DEVICE)
107108
math_mode!(new_handle, cuda.math_mode)
108109

109110
(; handle=new_handle, cuda.stream, cuda.math_mode)

lib/cublas/libcublas.jl

Lines changed: 602 additions & 602 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)