From 38ee7c93011946ac3cb5231600e09a0a583d93b0 Mon Sep 17 00:00:00 2001 From: Ye Tao Date: Mon, 3 Mar 2025 11:30:45 +0000 Subject: [PATCH] Add dispatch of SBGEMVNKERNEL for NEOVERSEN2 and NEOVERSEV2 --- kernel/arm64/KERNEL.NEOVERSEN2 | 3 ++- kernel/arm64/KERNEL.NEOVERSEV2 | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/kernel/arm64/KERNEL.NEOVERSEN2 b/kernel/arm64/KERNEL.NEOVERSEN2 index e4e1cfde3..fc7fe6930 100644 --- a/kernel/arm64/KERNEL.NEOVERSEN2 +++ b/kernel/arm64/KERNEL.NEOVERSEN2 @@ -198,4 +198,5 @@ SBGEMMINCOPYOBJ = sbgemm_incopy$(TSUFFIX).$(SUFFIX) SBGEMMITCOPYOBJ = sbgemm_itcopy$(TSUFFIX).$(SUFFIX) SBGEMMONCOPYOBJ = sbgemm_oncopy$(TSUFFIX).$(SUFFIX) SBGEMMOTCOPYOBJ = sbgemm_otcopy$(TSUFFIX).$(SUFFIX) -SBGEMVTKERNEL = sbgemv_t_bfdot.c \ No newline at end of file +SBGEMVTKERNEL = sbgemv_t_bfdot.c +SBGEMVNKERNEL = sbgemv_n_neon.c \ No newline at end of file diff --git a/kernel/arm64/KERNEL.NEOVERSEV2 b/kernel/arm64/KERNEL.NEOVERSEV2 index 4d866f858..e08efdb9d 100644 --- a/kernel/arm64/KERNEL.NEOVERSEV2 +++ b/kernel/arm64/KERNEL.NEOVERSEV2 @@ -2,4 +2,5 @@ include $(KERNELDIR)/KERNEL.ARMV8SVE ifeq ($(BUILD_BFLOAT16), 1) SBGEMVTKERNEL = sbgemv_t_bfdot.c +SBGEMVNKERNEL = sbgemv_n_neon.c endif \ No newline at end of file