From 4297e2ed84bbb265beca3b8619f8ca16b2d0479f Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Wed, 29 Apr 2020 22:09:23 +0200 Subject: [PATCH] fix shgemm parameter references in arm64 branch --- kernel/setparam-ref.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/kernel/setparam-ref.c b/kernel/setparam-ref.c index bf000f052..1ddab65ab 100644 --- a/kernel/setparam-ref.c +++ b/kernel/setparam-ref.c @@ -707,19 +707,25 @@ gotoblas_t TABLE_NAME = { #if defined(ARCH_ARM64) static void init_parameter(void) { +#if defined(BUILD_HALFi) TABLE_NAME.shgemm_p = SHGEMM_DEFAULT_P; +#endif TABLE_NAME.sgemm_p = SGEMM_DEFAULT_P; TABLE_NAME.dgemm_p = DGEMM_DEFAULT_P; TABLE_NAME.cgemm_p = CGEMM_DEFAULT_P; TABLE_NAME.zgemm_p = ZGEMM_DEFAULT_P; +#if defined(BUILD_HALF) TABLE_NAME.shgemm_q = SHGEMM_DEFAULT_Q; +#endif TABLE_NAME.sgemm_q = SGEMM_DEFAULT_Q; TABLE_NAME.dgemm_q = DGEMM_DEFAULT_Q; TABLE_NAME.cgemm_q = CGEMM_DEFAULT_Q; TABLE_NAME.zgemm_q = ZGEMM_DEFAULT_Q; +#if defined(BUILD_HALF) TABLE_NAME.shgemm_r = SHGEMM_DEFAULT_R; +#endif TABLE_NAME.sgemm_r = SGEMM_DEFAULT_R; TABLE_NAME.dgemm_r = DGEMM_DEFAULT_R; TABLE_NAME.cgemm_r = CGEMM_DEFAULT_R;