debug

Author: amontoison

src/solver/highlevel.jl

@@ -226,9 +226,9 @@ for (fname, elty) in (
             (mA == nA == mB == nB == mC == nC) || throw(DimensionMismatch("The first two dimensions of A, B and C must match"))
             (nblocksA == nblocksB - 1 == nblocksC) || throw(DimensionMismatch("Inconsistency for the last dimension of A, B and C"))
 
-            lda = max(1, stride(A, 3))
-            ldb = max(1, stride(B, 3))
-            ldc = max(1, stride(C, 3))
+            lda = max(1, stride(A, 2))
+            ldb = max(1, stride(B, 2))
+            ldc = max(1, stride(C, 2))
 
             devinfo = ROCArray{Cint}(undef, 1)
             $fname(rocBLAS.handle(), mB, nblocksB, A, lda, B, ldb, C, ldc, devinfo)
@@ -252,15 +252,14 @@ for (fname, elty) in (
             mB, nB, nblocksB = size(B)
             mC, nC, nblocksC = size(C)
             mX, nblocksX, nrhs = size(X)
-            nrhs = size(X, 2)
             (mA == nA == mB == nB == mC == nC) || throw(DimensionMismatch("The first two dimensions of A, B and C must match"))
             (mX == mA) || throw(DimensionMismatch("The first dimension of X is inconsistent with first two dimensions of A, B and C"))
             (nblocksA == nblocksB - 1 == nblocksX - 1 == nblocksC) || throw(DimensionMismatch("Inconsistency for the number of blocks in A, B, C and X"))
 
-            lda = max(1, stride(A, 3))
-            ldb = max(1, stride(B, 3))
-            ldc = max(1, stride(C, 3))
-            ldx = max(1, stride(X, 3))
+            lda = max(1, stride(A, 2))
+            ldb = max(1, stride(B, 2))
+            ldc = max(1, stride(C, 2))
+            ldx = max(1, stride(X, 2))
 
             $fname(rocBLAS.handle(), mB, nblocksB, nrhs, A, lda, B, ldb, C, ldc, X, ldx)
             X