From 0b663a112e1fc6fdcc1915423b752325e3d5452e Mon Sep 17 00:00:00 2001 From: ajz34 Date: Sat, 15 Aug 2026 22:45:20 +0800 Subject: [PATCH] Fix gemm_batch small matrix kernel address retrieval for correct call with DYNAMIC_ARCH activated Co-authored-by: Claude Code Co-authored-by: glm-5.3 --- interface/gemm_batch.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/interface/gemm_batch.c b/interface/gemm_batch.c index 16b6b2f401..7a8789e820 100644 --- a/interface/gemm_batch.c +++ b/interface/gemm_batch.c @@ -353,18 +353,18 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE * transa_array, enum CB #if !defined(COMPLEX) if(*(FLOAT *)(group_beta) == 0.0){ group_mode=mode | BLAS_SMALL_B0_OPT; - group_small_matrix_opt_routine=(void *)(gemm_small_kernel_b0[(group_transb<<2)|group_transa]); + group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(gemm_small_kernel_b0, ((group_transb<<2)|group_transa)); }else{ group_mode=mode | BLAS_SMALL_OPT; - group_small_matrix_opt_routine=(void *)(gemm_small_kernel[(group_transb<<2)|group_transa]); + group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(gemm_small_kernel, ((group_transb<<2)|group_transa)); } #else if(((FLOAT *)(group_beta))[0] == 0.0 && ((FLOAT *)(group_beta))[1] == 0.0){ group_mode=mode | BLAS_SMALL_B0_OPT; - group_small_matrix_opt_routine=(void *)(zgemm_small_kernel_b0[(group_transb<<2)|group_transa]); + group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(zgemm_small_kernel_b0, ((group_transb<<2)|group_transa)); }else{ group_mode=mode | BLAS_SMALL_OPT; - group_small_matrix_opt_routine=(void *)(zgemm_small_kernel[(group_transb<<2)|group_transa]); + group_small_matrix_opt_routine=SMALL_KERNEL_ADDR(zgemm_small_kernel, ((group_transb<<2)|group_transa)); } #endif