From 61b5a56a3a4acd34f23307172788dd4d76c9e30b Mon Sep 17 00:00:00 2001 From: Katharine Hyatt Date: Mon, 1 Jun 2026 11:40:45 +0200 Subject: [PATCH] Test QR rules with CUDA --- test/mooncake/qr.jl | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/test/mooncake/qr.jl b/test/mooncake/qr.jl index bbb9a8d17..c4f0df9e0 100644 --- a/test/mooncake/qr.jl +++ b/test/mooncake/qr.jl @@ -20,4 +20,11 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23) TestSuite.test_mooncake_qr(AT, (m, m); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T)) end end + if T ∈ BLASFloats && CUDA.functional() + TestSuite.test_mooncake_qr(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T)) + if m == n + AT = Diagonal{T, CuVector{T}} + TestSuite.test_mooncake_qr(AT, (m, m); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T)) + end + end end