Skip to content

Commit 5bd5547

Browse files
authored
Merge pull request #5979 from ajz34/fix/strided_colmajor
Fix: CBLAS order handling in `gemm_batch_strided` for column-major layout
2 parents b08df58 + 86c65a5 commit 5bd5547

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

interface/gemm_batch_strided.c

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -229,13 +229,16 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE transa, enum CBLAS_TRANS
229229
group_transb = -1;
230230
info = 0;
231231

232-
#if defined(CBLAS)
232+
#if defined(CBLAS)
233233
if (order == CblasColMajor) {
234234
group_m = m;
235235
group_n = n;
236236
group_k = k;
237237

238-
238+
group_lda = lda;
239+
group_ldb = ldb;
240+
group_ldc = ldc;
241+
239242
if (transa == CblasNoTrans) group_transa = 0;
240243
if (transa == CblasTrans) group_transa = 1;
241244
#ifndef COMPLEX

0 commit comments

Comments
 (0)