Fix transposed lda on cache

EnzymeAD · Oct 24, 2023 · 4e5639c · 4e5639c
1 parent 9f7cb46
commit 4e5639c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -2597,7 +2597,7 @@ llvm::Value *transpose(IRBuilder<> &B, llvm::Value *V, bool cublas) {
 
 // Implement the following logic to get the width of a matrix
 // if (cache_A) {
-//   ld_A = (arg_transa == 'N') ? arg_m : arg_k;
+//   ld_A = (arg_transa == 'N') ? arg_k : arg_m;
 // } else {
 //   ld_A = arg_lda;
 // }

diff --git a/enzyme/test/Integration/ReverseMode/blas.cpp b/enzyme/test/Integration/ReverseMode/blas.cpp
@@ -243,7 +243,7 @@ static void gemvTests() {
 							lda);
 
             // dB = alpha * trans(A) * dC + dB
-            cblas_dgemv(layout, transpose(transA), M, N, alpha, A_cache, M, dC, incC, 1.0, dB, incB); 
+            cblas_dgemv(layout, transpose(transA), M, N, alpha, A_cache, N, dC, incC, 1.0, dB, incB); 
 
             // dY = beta * dY
             cblas_dscal(trans ? N : M, beta, dC, incC);