LoongArch: Small matrix opt

OpenMathLib · May 21, 2024 · ff1e7eb · ff1e7eb
1 parent 700ea74
commit ff1e7eb
Show file tree

Hide file tree

Showing 4 changed files with 593 additions and 0 deletions.
diff --git a/Makefile.system b/Makefile.system
@@ -269,6 +269,7 @@ else ifeq ($(ARCH), power)
 SMALL_MATRIX_OPT = 1
 BUILD_BFLOAT16 = 1
 endif
+SMALL_MATRIX_OPT = 1
 ifeq ($(SMALL_MATRIX_OPT), 1)
 CCOMMON_OPT += -DSMALL_MATRIX_OPT
 endif

diff --git a/kernel/loongarch64/KERNEL.LOONGSON3R5 b/kernel/loongarch64/KERNEL.LOONGSON3R5
@@ -162,4 +162,8 @@ STRSMKERNEL_LN  = ../generic/trsm_kernel_LN.c
 STRSMKERNEL_LT  = ../generic/trsm_kernel_LT.c
 STRSMKERNEL_RN  = ../generic/trsm_kernel_RN.c
 STRSMKERNEL_RT  = ../generic/trsm_kernel_RT.c
+
+DGEMM_SMALL_M_PERMIT = dgemm_small_matrix_permit.c
+DGEMM_SMALL_K_NN = dgemm_small_kernel_nn_lasx.S
+DGEMM_SMALL_K_B0_NN = dgemm_small_kernel_nn_lasx.S
 endif