Let's try with BigFloat

Author: kshyatt

test/mooncake.jl

@@ -5,14 +5,14 @@ using CUDA, AMDGPU
 
 #BLASFloats = (Float32, Float64, ComplexF32, ComplexF64)
 BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
-
+GenericFloats = (BigFloat,)
 @isdefined(TestSuite) || include("testsuite/TestSuite.jl")
 using .TestSuite
 
 is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
 
 m = 19
-for T in BLASFloats, n in (17, m, 23)
+for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     TestSuite.seed_rng!(123)
     if CUDA.functional()
         TestSuite.test_mooncake(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))

test/testsuite/mooncake.jl

@@ -3,6 +3,8 @@ using MatrixAlgebraKit
 using Mooncake, Mooncake.TestUtils
 using Mooncake: rrule!!
 using MatrixAlgebraKit: diagview, TruncatedAlgorithm, PolarViaSVD, eigh_trunc
+using LinearAlgebra: BlasFloat
+using GenericLinearAlgebra
 
 function mc_copy_eigh_full(A; kwargs...)
     A = (A + A') / 2
@@ -160,7 +162,7 @@ function test_mooncake(T::Type, sz; kwargs...)
         test_mooncake_qr(T, sz; kwargs...)
         test_mooncake_lq(T, sz; kwargs...)
         if length(sz) == 1 || sz[1] == sz[2]
-            test_mooncake_eig(T, sz; kwargs...)
+            T <: BlasFloat && test_mooncake_eig(T, sz; kwargs...)
             test_mooncake_eigh(T, sz; kwargs...)
         end
         test_mooncake_svd(T, sz; kwargs...)