Skip to content

Commit abf0200

Browse files
committed
Fix tri-tri mul tests
1 parent 527356a commit abf0200

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

test/libraries/cublas/level3_gemm.jl

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -219,20 +219,20 @@ k = 13
219219
sB = rand(elty,m,m)
220220
sB = sB + transpose(sB)
221221

222-
for (TRa, ta, TRb, tb, TRc) in (
223-
(UpperTriangular, identity, LowerTriangular, identity, Matrix),
224-
(LowerTriangular, identity, UpperTriangular, identity, Matrix),
225-
(UpperTriangular, identity, UpperTriangular, transpose, Matrix),
226-
(UpperTriangular, transpose, UpperTriangular, identity, Matrix),
227-
(LowerTriangular, identity, LowerTriangular, transpose, Matrix),
228-
(LowerTriangular, transpose, LowerTriangular, identity, Matrix),
222+
for (TRa, ta, TRb, tb, TRc, a_func, b_func) in (
223+
(UpperTriangular, identity, LowerTriangular, identity, Matrix, triu, tril),
224+
(LowerTriangular, identity, UpperTriangular, identity, Matrix, tril, triu),
225+
(UpperTriangular, identity, UpperTriangular, transpose, Matrix, triu, triu),
226+
(UpperTriangular, transpose, UpperTriangular, identity, Matrix, triu, triu),
227+
(LowerTriangular, identity, LowerTriangular, transpose, Matrix, tril, tril),
228+
(LowerTriangular, transpose, LowerTriangular, identity, Matrix, tril, tril),
229229
)
230230

231231
A = copy(sA) |> TRa
232232
B = copy(sB) |> TRb
233233
C = copy(C0) |> TRc
234-
dA = CuArray(parent(sA)) |> TRa
235-
dB = CuArray(parent(sB)) |> TRb
234+
dA = CuArray(a_func(parent(sA))) |> TRa
235+
dB = CuArray(b_func(parent(sB))) |> TRb
236236
dC = if TRc == Matrix
237237
CuArray(C0) |> DenseCuMatrix
238238
else

0 commit comments

Comments
 (0)