diff --git a/src/blas/wrappers.jl b/src/blas/wrappers.jl index 53fc9d2af..4cf93e4be 100644 --- a/src/blas/wrappers.jl +++ b/src/blas/wrappers.jl @@ -809,7 +809,7 @@ for (fname, elty) in ((:rocblas_dsyrk,:Float64), @eval begin function syrk!( uplo::Char, trans::Char, alpha::($elty), - A::ROCVecOrMat{$elty}, beta::($elty), C::ROCMatrix{$elty}, + A::StridedROCVecOrMat{$elty}, beta::($elty), C::StridedROCMatrix{$elty}, ) mC, n = size(C) if mC != n throw(DimensionMismatch("C must be square")) end @@ -974,7 +974,7 @@ for (mmname, smname, elty) in @eval begin function trmm!( side::Char, uplo::Char, transa::Char, diag::Char, alpha::($elty), - A::ROCMatrix{$elty}, B::ROCMatrix{$elty}, C::ROCMatrix{$elty}, + A::StridedROCMatrix{$elty}, B::StridedROCMatrix{$elty}, C::StridedROCMatrix{$elty}, ) m, n = size(B) mA, nA = size(A) @@ -992,13 +992,13 @@ for (mmname, smname, elty) in end function trmm( side::Char, uplo::Char, transa::Char, diag::Char, alpha::($elty), - A::ROCMatrix{$elty}, B::ROCMatrix{$elty}, + A::StridedROCMatrix{$elty}, B::StridedROCMatrix{$elty}, ) trmm!(side, uplo, transa, diag, alpha, A, B, similar(B)) end function trsm!( side::Char, uplo::Char, transa::Char, diag::Char, alpha::($elty), - A::ROCMatrix{$elty}, B::ROCMatrix{$elty}, + A::StridedROCMatrix{$elty}, B::StridedROCMatrix{$elty}, ) m, n = size(B) mA, nA = size(A) @@ -1013,7 +1013,7 @@ for (mmname, smname, elty) in end function trsm( side::Char, uplo::Char, transa::Char, diag::Char, alpha::($elty), - A::ROCMatrix{$elty}, B::ROCMatrix{$elty}, + A::StridedROCMatrix{$elty}, B::StridedROCMatrix{$elty}, ) trsm!(side, uplo, transa, diag, alpha, A, copy(B)) end diff --git a/src/solver/highlevel.jl b/src/solver/highlevel.jl index 5c3a5d048..eb07b2368 100644 --- a/src/solver/highlevel.jl +++ b/src/solver/highlevel.jl @@ -5,7 +5,7 @@ for (fname, elty) in ( (:rocsolver_zpotrf, :ComplexF64), ) @eval begin - function potrf!(uplo::Char, A::ROCMatrix{$elty}) + function potrf!(uplo::Char, A::StridedROCMatrix{$elty}) chkuplo(uplo) n = checksquare(A) lda = max(1, stride(A, 2))