More GPU fixes

Author: kshyatt

Project.toml

@@ -57,3 +57,6 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
 test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "Random", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Mooncake"]
+
+[sources]
+CUDA = {url="https://github.com/JuliaGPU/CUDA.jl", rev="master"}

test/testsuite/ad_utils.jl

@@ -29,8 +29,8 @@ function remove_eighgauge_dependence!(
 end
 
 function stabilize_eigvals!(D::AbstractVector)
-    absD = abs.(D)
-    p = invperm(sortperm(absD)) # rank of abs(D)
+    absD = collect(abs.(D))
+    p = invperm(sortperm(collect(absD))) # rank of abs(D)
     # account for exact degeneracies in absolute value when having complex conjugate pairs
     for i in 1:(length(D) - 1)
         if absD[i] == absD[i + 1] # conjugate pairs will appear sequentially
@@ -41,9 +41,7 @@ function stabilize_eigvals!(D::AbstractVector)
     # rescale eigenvalues so that they lie on distinct radii in the complex plane
     # that are chosen randomly in non-overlapping intervals [k/n, (k+0.5)/n)] for k=1,...,n
     radii = ((1:n) .+ rand(real(eltype(D)), n) ./ 2) ./ n
-    for i in 1:length(D)
-        D[i] = sign(D[i]) * radii[p[i]]
-    end
+    D .= sign.(D) .* radii[p]
     return D
 end
 function make_eig_matrix(T, sz)