Browse Source

declare DGEMM_BETA in KERNEL.ARMV8 rather than the generic KERNEL

tags/v0.3.8^2
w00421467 6 years ago
parent
commit
b7cc69ee62
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      kernel/arm64/KERNEL
  2. +2
    -0
      kernel/arm64/KERNEL.ARMV8

+ 1
- 1
kernel/arm64/KERNEL View File

@@ -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


+ 2
- 0
kernel/arm64/KERNEL.ARMV8 View File

@@ -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))


Loading…
Cancel
Save