Skip to content

Commit f2ba6a2

Browse files
authored
Merge pull request #5980 from ajz34/fix/dynarch_small_matrix_batch
Fix: Small matrix `gemm_batch` with DYNAMIC_ARCH
2 parents 5bd5547 + 0b663a1 commit f2ba6a2

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

interface/gemm_batch.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -353,18 +353,18 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE * transa_array, enum CB
353353
#if !defined(COMPLEX)
354354
if(*(FLOAT *)(group_beta) == 0.0){
355355
group_mode=mode | BLAS_SMALL_B0_OPT;
356-
group_small_matrix_opt_routine=(void *)(gemm_small_kernel_b0[(group_transb<<2)|group_transa]);
356+
group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(gemm_small_kernel_b0, ((group_transb<<2)|group_transa));
357357
}else{
358358
group_mode=mode | BLAS_SMALL_OPT;
359-
group_small_matrix_opt_routine=(void *)(gemm_small_kernel[(group_transb<<2)|group_transa]);
359+
group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(gemm_small_kernel, ((group_transb<<2)|group_transa));
360360
}
361361
#else
362362
if(((FLOAT *)(group_beta))[0] == 0.0 && ((FLOAT *)(group_beta))[1] == 0.0){
363363
group_mode=mode | BLAS_SMALL_B0_OPT;
364-
group_small_matrix_opt_routine=(void *)(zgemm_small_kernel_b0[(group_transb<<2)|group_transa]);
364+
group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(zgemm_small_kernel_b0, ((group_transb<<2)|group_transa));
365365
}else{
366366
group_mode=mode | BLAS_SMALL_OPT;
367-
group_small_matrix_opt_routine=(void *)(zgemm_small_kernel[(group_transb<<2)|group_transa]);
367+
group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(zgemm_small_kernel, ((group_transb<<2)|group_transa));
368368
}
369369

370370
#endif

0 commit comments

Comments
 (0)