Modification to prevent scalar indexing on CUDA devices

Scalar indexing on CUDA devices results in poor performance. This modification uses dot syntax to prevent scalar indexing.
JuliaLinearAlgebra · May 16, 2024 · eef5bd0 · eef5bd0
1 parent e5d10e2
commit eef5bd0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/lsqr.jl b/src/lsqr.jl
@@ -98,10 +98,12 @@ function lsqr_method!(log::ConvergenceHistory, x, A, b;
     n = size(A,2)
     length(x) == n || error("x should be of length ", n)
     length(b) == m || error("b should be of length ", m)
-    for i = 1:n
-        isfinite(x[i]) || error("Initial guess for x must be finite")
+
+    if !(all(isfinite.(x)))
+      error("Initial guess for x must be finite")
     end
 
+
     # Initialize
     T = Adivtype(A, b)
     Tr = real(T)