@@ -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