Skip to content

Commit 61b5a56

Browse files
committed
Test QR rules with CUDA
1 parent 08eaf37 commit 61b5a56

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

test/mooncake/qr.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,11 @@ for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
2020
TestSuite.test_mooncake_qr(AT, (m, m); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
2121
end
2222
end
23+
if T BLASFloats && CUDA.functional()
24+
TestSuite.test_mooncake_qr(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
25+
if m == n
26+
AT = Diagonal{T, CuVector{T}}
27+
TestSuite.test_mooncake_qr(AT, (m, m); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
28+
end
29+
end
2330
end

0 commit comments

Comments
 (0)