Merge pull request #362 from oscarvanderheide/ovdh_fix_scalar_indexing

Modification to prevent scalar indexing on CUDA devices

Author: ViralBShah

Diff for: 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)