Browse Source

Add dispatch of SBGEMVNKERNEL for NEOVERSEN2 and NEOVERSEV2

tags/v0.3.30
Ye Tao 8 months ago
parent
commit
38ee7c9301
2 changed files with 3 additions and 1 deletions
  1. +2
    -1
      kernel/arm64/KERNEL.NEOVERSEN2
  2. +1
    -0
      kernel/arm64/KERNEL.NEOVERSEV2

+ 2
- 1
kernel/arm64/KERNEL.NEOVERSEN2 View File

@@ -198,4 +198,5 @@ SBGEMMINCOPYOBJ = sbgemm_incopy$(TSUFFIX).$(SUFFIX)
SBGEMMITCOPYOBJ = sbgemm_itcopy$(TSUFFIX).$(SUFFIX) SBGEMMITCOPYOBJ = sbgemm_itcopy$(TSUFFIX).$(SUFFIX)
SBGEMMONCOPYOBJ = sbgemm_oncopy$(TSUFFIX).$(SUFFIX) SBGEMMONCOPYOBJ = sbgemm_oncopy$(TSUFFIX).$(SUFFIX)
SBGEMMOTCOPYOBJ = sbgemm_otcopy$(TSUFFIX).$(SUFFIX) SBGEMMOTCOPYOBJ = sbgemm_otcopy$(TSUFFIX).$(SUFFIX)
SBGEMVTKERNEL = sbgemv_t_bfdot.c
SBGEMVTKERNEL = sbgemv_t_bfdot.c
SBGEMVNKERNEL = sbgemv_n_neon.c

+ 1
- 0
kernel/arm64/KERNEL.NEOVERSEV2 View File

@@ -2,4 +2,5 @@ include $(KERNELDIR)/KERNEL.ARMV8SVE


ifeq ($(BUILD_BFLOAT16), 1) ifeq ($(BUILD_BFLOAT16), 1)
SBGEMVTKERNEL = sbgemv_t_bfdot.c SBGEMVTKERNEL = sbgemv_t_bfdot.c
SBGEMVNKERNEL = sbgemv_n_neon.c
endif endif

Loading…
Cancel
Save