From c43e428797eda78c9a43c4956d8626417f207abb Mon Sep 17 00:00:00 2001 From: Katharine Hyatt Date: Mon, 1 Jun 2026 11:39:10 +0200 Subject: [PATCH] Test Mooncake projection rules for CUDA --- test/mooncake/projections.jl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/mooncake/projections.jl b/test/mooncake/projections.jl index b33967e9d..5e2a1e2c5 100644 --- a/test/mooncake/projections.jl +++ b/test/mooncake/projections.jl @@ -18,4 +18,8 @@ for T in (BLASFloats..., GenericFloats...) TestSuite.test_mooncake_projections(T, (m, m); atol, rtol) TestSuite.test_mooncake_projections(Diagonal{T, Vector{T}}, (m, m); atol, rtol) end + if T ∈ BLASFloats && CUDA.functional() + TestSuite.test_mooncake_projections(CuMatrix{T}, (m, m); atol, rtol) + TestSuite.test_mooncake_projections(Diagonal{T, CuVector{T}}, (m, m); atol, rtol) + end end