Skip to content

Commit dd2332b

Browse files
Add view of potrf test
1 parent 1306be5 commit dd2332b

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

test/rocarray/solver.jl

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,21 @@ end
183183
LAPACK.potrs!('U',A,B)
184184
@test B collect(d_B)
185185
end
186+
@testset "elty = $elty strided" for elty in [Float32, Float64, ComplexF32, ComplexF64]
187+
A = rand(elty,n*2,n*2)
188+
A = A*A' + I
189+
B = rand(elty,n,p)
190+
d_A = view(ROCArray(A),1:n,1:n)
191+
d_B = ROCArray(B)
192+
193+
LAPACK.potrf!('L',d_A)
194+
LAPACK.potrs!('U',copy(d_A),d_B)
195+
LAPACK.potrf!('L',view(A,1:n,1:n))
196+
LAPACK.potrs!('U',copy(view(A,1:n,1:n)),B)
197+
@test B collect(d_B)
198+
end
199+
200+
186201
end
187202

188203
@testset "sytrf!" begin

0 commit comments

Comments
 (0)