Update gpu.jl

JuliaSmoothOptimizers · Nov 30, 2023 · 8917a13 · 8917a13
1 parent 0ccd647
commit 8917a13
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/gpu/gpu.jl b/test/gpu/gpu.jl
@@ -74,14 +74,14 @@ function test_operator(FC, V, DM, SM)
     mul!(y_gpu, opA_gpu, x_gpu)
     @test collect(y_gpu) ≈ y_cpu
   end
-  for j = 1:5
+  for j = 10:10:50
     y_cpu = rand(FC, m)
     x_cpu = rand(FC, n)
-    A_cpu2 = A_cpu + j*I
+    A_cpu2 = j*A_cpu
     mul!(y_cpu, A_cpu2, x_cpu)
     y_gpu = V(y_cpu)
     x_gpu = V(x_cpu)
-    A_gpu2 = A_gpu + j*I
+    A_gpu2 = j*A_gpu
     update_operator!(opA_gpu, A_gpu2)
     mul!(y_gpu, opA_gpu, x_gpu)
     @test collect(y_gpu) ≈ y_cpu
@@ -98,14 +98,14 @@ function test_operator(FC, V, DM, SM)
     mul!(Y_gpu, opA_gpu, X_gpu)
     @test collect(Y_gpu) ≈ Y_cpu
   end
-  for j = 1:5
+  for j = 10:10:50
     Y_cpu = rand(FC, m, nrhs)
     X_cpu = rand(FC, n, nrhs)
-    A_cpu2 = A_cpu + j*I
+    A_cpu2 = j*A_cpu
     mul!(Y_cpu, A_cpu2, X_cpu)
     Y_gpu = DM(Y_cpu)
     X_gpu = DM(X_cpu)
-    A_gpu2 = A_gpu + j*I
+    A_gpu2 = j*A_gpu
     update_operator!(opA_gpu, A_gpu2)
     mul!(Y_gpu, opA_gpu, X_gpu)
     @test collect(Y_gpu) ≈ Y_cpu