Skip to content

Commit 5f924f2

Browse files
authored
Test Mooncake projection rules for CUDA (#240)
1 parent 08eaf37 commit 5f924f2

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

test/mooncake/projections.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,8 @@ for T in (BLASFloats..., GenericFloats...)
1818
TestSuite.test_mooncake_projections(T, (m, m); atol, rtol)
1919
TestSuite.test_mooncake_projections(Diagonal{T, Vector{T}}, (m, m); atol, rtol)
2020
end
21+
if T BLASFloats && CUDA.functional()
22+
TestSuite.test_mooncake_projections(CuMatrix{T}, (m, m); atol, rtol)
23+
TestSuite.test_mooncake_projections(Diagonal{T, CuVector{T}}, (m, m); atol, rtol)
24+
end
2125
end

0 commit comments

Comments
 (0)