refactor projections tests

Author: lkdvos

test/mooncake/projections.jl

@@ -0,0 +1,19 @@
+using MatrixAlgebraKit
+using Test
+using LinearAlgebra: Diagonal
+using CUDA, AMDGPU
+
+BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
+GenericFloats = ()
+@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
+using .TestSuite
+
+is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
+
+m = 19
+for T in (BLASFloats..., GenericFloats...)
+    TestSuite.seed_rng!(123)
+    if !is_buildkite
+        TestSuite.test_mooncake_projections(T, (m, m); atol = m * m * TestSuite.precision(T), rtol = m * m * TestSuite.precision(T))
+    end
+end

test/testsuite/TestSuite.jl

@@ -87,8 +87,6 @@ instantiate_unitary(::Type{<:Diagonal}, A, sz) = Diagonal(fill!(similar(parent(A
 
 include("ad_utils.jl")
 
-include("projections.jl")
-
 # Decompositions
 # --------------
 include("decompositions/qr.jl")
@@ -110,6 +108,7 @@ include("mooncake/eigh.jl")
 include("mooncake/svd.jl")
 include("mooncake/polar.jl")
 include("mooncake/orthnull.jl")
+include("mooncake/projections.jl")
 
 include("enzyme.jl")
 include("chainrules.jl")