[documentation] Use CUDA.zeros instead of similar

JuliaSmoothOptimizers · Mar 31, 2023 · 19542fa · 19542fa
1 parent c1be270
commit 19542fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docs/src/gpu.md b/docs/src/gpu.md
@@ -70,7 +70,7 @@ if CUDA.functional()
   # Additional vector required for solving triangular systems
   n = length(b_gpu)
   T = eltype(b_gpu)
-  z = similar(CuVector{T}, n)
+  z = CUDA.zeros(T, n)
 
   # Solve Py = x
   function ldiv_ic0!(P::CuSparseMatrixCSR, x, y, z)
@@ -116,7 +116,7 @@ if CUDA.functional()
   # Additional vector required for solving triangular systems
   n = length(b_gpu)
   T = eltype(b_gpu)
-  z = similar(CuVector{T}, n)
+  z = CUDA.zeros(T, n)
 
   # Solve Py = x
   function ldiv_ilu0!(P::CuSparseMatrixCSR, x, y, z)

diff --git a/test/gpu/nvidia.jl b/test/gpu/nvidia.jl
@@ -27,7 +27,7 @@ include("gpu.jl")
       b_gpu = CuVector(b_cpu)
       n = length(b_gpu)
       T = eltype(b_gpu)
-      z = similar(CuVector{T}, n)
+      z = CUDA.zeros(T, n)
       symmetric = hermitian = true
 
       A_gpu = CuSparseMatrixCSC(A_cpu)
@@ -72,7 +72,7 @@ include("gpu.jl")
       b_gpu = CuVector(b_cpu)
       n = length(b_gpu)
       T = eltype(b_gpu)
-      z = similar(CuVector{T}, n)
+      z = CUDA.zeros(T, n)
       symmetric = hermitian = false
 
       A_gpu = CuSparseMatrixCSC(A_cpu[:,p])