Skip to content

Commit 95e7822

Browse files
committed
Fix gemm_batch small matrix kernel address retrieval for correct call with DYNAMIC_ARCH activated
1 parent 05518e0 commit 95e7822

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)