From b7cc69ee622fed9039ab755b87eee9279d27d541 Mon Sep 17 00:00:00 2001 From: w00421467 Date: Fri, 20 Dec 2019 10:11:50 +0800 Subject: [PATCH] declare DGEMM_BETA in KERNEL.ARMV8 rather than the generic KERNEL --- kernel/arm64/KERNEL | 2 +- kernel/arm64/KERNEL.ARMV8 | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/kernel/arm64/KERNEL b/kernel/arm64/KERNEL index 4402571..f936cdf 100644 --- a/kernel/arm64/KERNEL +++ b/kernel/arm64/KERNEL @@ -34,7 +34,7 @@ ifndef SGEMM_BETA SGEMM_BETA = ../generic/gemm_beta.c endif ifndef DGEMM_BETA -DGEMM_BETA = ../arm64/dgemm_beta.S +DGEMM_BETA = ../generic/gemm_beta.c endif ifndef CGEMM_BETA CGEMM_BETA = ../generic/zgemm_beta.c diff --git a/kernel/arm64/KERNEL.ARMV8 b/kernel/arm64/KERNEL.ARMV8 index efc1ec8..b90dd22 100644 --- a/kernel/arm64/KERNEL.ARMV8 +++ b/kernel/arm64/KERNEL.ARMV8 @@ -102,6 +102,8 @@ CDOTKERNEL = zdot.S ZDOTKERNEL = zdot.S DSDOTKERNEL = dot.S +DGEMM_BETA = dgemm_beta.S + SGEMMKERNEL = sgemm_kernel_$(SGEMM_UNROLL_M)x$(SGEMM_UNROLL_N).S STRMMKERNEL = strmm_kernel_$(SGEMM_UNROLL_M)x$(SGEMM_UNROLL_N).S ifneq ($(SGEMM_UNROLL_M), $(SGEMM_UNROLL_N)) -- 2.7.4