scalar indexing fix

Author: thomvet

test/gpu_all.jl

@@ -48,24 +48,22 @@ prob = ODEProblem(foo, u0, (0.0f0,1.0f0), (A, cache))
 sol = solve(prob, TRBDF2())
 @test sol.retcode == :Success
 
+#resizing tests
 randmat = cu(rand(5, 3))
 sto = similar(randmat)
 stod = dualcache(sto)
 function claytonsample!(sto, τ, α; randmat=randmat)
     sto = get_tmp(sto, τ)
-        sto .= randmat
+    sto .= randmat
     τ == 0 && return sto
     n = size(sto, 1)
-    for i in 1:n
-        v = sto[i, 2]
-        u = sto[i, 1]
-        sto[i, 1] = (1 - u^(-τ) + u^(-τ)*v^(-(τ/(1 + τ))))^(-1/τ)*α
-        sto[i, 2] = (1 - u^(-τ) + u^(-τ)*v^(-(τ/(1 + τ))))^(-1/τ)
-    end
+    v = @view sto[:, 2]
+    u = @view sto[:, 1]
+    @. v = (1 - u^(-τ) + u^(-τ)*v^(-(τ/(1 + τ))))^(-1/τ)*α
+    @. u = (1 - u^(-τ) + u^(-τ)*v^(-(τ/(1 + τ))))^(-1/τ)
     return sto
 end
 
-#resizing tests
 #taking the derivative of claytonsample! with respect to τ only
 df1 = ForwardDiff.derivative(τ -> claytonsample!(stod, τ, 0.0), 0.3)
 @test size(randmat) == size(df1)