Skip to content

Commit d07f57d

Browse files
committed
Specialize for Diagonal of ROCVector
1 parent e4c8ebb commit d07f57d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/testsuite/TestSuite.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,11 @@ is_pivoted(alg::MatrixAlgebraKit.LQViaTransposedQR) = is_pivoted(alg.qr_alg)
7272
isleftcomplete(V, N) = V * V' + N * N' I
7373
isleftcomplete(V::AnyCuMatrix, N::AnyCuMatrix) = isleftcomplete(collect(V), collect(N))
7474
isleftcomplete(V::AnyROCMatrix, N::AnyROCMatrix) = isleftcomplete(collect(V), collect(N))
75+
isleftcomplete(V::Diagonal{T, <:ROCVector{<:T}}, N::Diagonal{T, <:ROCVector{<:T}}) where {T} = isleftcomplete(collect(V), collect(N))
7576
isrightcomplete(Vᴴ, Nᴴ) = Vᴴ' * Vᴴ + Nᴴ' * Nᴴ I
7677
isrightcomplete(V::AnyCuMatrix, N::AnyCuMatrix) = isrightcomplete(collect(V), collect(N))
7778
isrightcomplete(V::AnyROCMatrix, N::AnyROCMatrix) = isrightcomplete(collect(V), collect(N))
79+
isrightcomplete(V::Diagonal{T, <:ROCVector{<:T}}, N::Diagonal{T, <:ROCVector{<:T}}) where {T} = isrightcomplete(collect(V), collect(N))
7880

7981
instantiate_unitary(T, A, sz) = qr_compact(randn!(similar(A, eltype(T), sz, sz)))[1]
8082
# AMDGPU can't generate ComplexF32 random numbers

0 commit comments

Comments
 (0)