Try Enzyme tests with CUDA

Author: kshyatt

test/enzyme/eig.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...)
     if !is_buildkite
         TestSuite.test_enzyme_eig(T, (m, m); atol = m * m * TestSuite.precision(T), rtol = m * m * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_eig(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/eigh.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...)
     if !is_buildkite
         TestSuite.test_enzyme_eigh(T, (m, m); atol = m * m * TestSuite.precision(T), rtol = m * m * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_eigh(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/lq.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     if !is_buildkite
         TestSuite.test_enzyme_lq(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_lq(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/orthnull.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     if !is_buildkite
         TestSuite.test_enzyme_orthnull(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_orthnull(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/polar.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     if !is_buildkite
         TestSuite.test_enzyme_polar(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_polar(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/projections.jl

@@ -18,4 +18,8 @@ for T in (BLASFloats..., GenericFloats...)
         TestSuite.test_enzyme_projections(T, (m, m); atol, rtol)
         TestSuite.test_enzyme_projections(Diagonal{T, Vector{T}}, (m, m); atol, rtol)
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_projections(CuMatrix{T}, (m, n); atol, rtol)
+        TestSuite.test_enzyme_projections(Diagonal{T, CuVector{T}}, (m, m); atol, rtol)
+    end
 end

test/enzyme/qr.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     if !is_buildkite
         TestSuite.test_enzyme_qr(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_qr(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end

test/enzyme/svd.jl

@@ -16,4 +16,7 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
     if !is_buildkite
         TestSuite.test_enzyme_svd(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
+    if CUDA.functional()
+        TestSuite.test_enzyme_svd(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
 end