Merge pull request OpenMathLib#4831 from martin-frbg/gemmforward

Enable forwarding from GEMM to GEMV for RISCV and PPC in addition to ARM64
martin-frbg · Aug 3, 2024 · 2c2b6bc · 2c2b6bc
2 parents 9afd0c8 + 42d8865
commit 2c2b6bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/Makefile.system b/Makefile.system
@@ -277,6 +277,12 @@ endif
 ifeq ($(ARCH), arm64)
 GEMM_GEMV_FORWARD = 1
 endif
+ifeq ($(ARCH), riscv)
+GEMM_GEMV_FORWARD = 1
+endif
+ifeq ($(ARCH), power)
+GEMM_GEMV_FORWARD = 1
+endif
 
 ifeq ($(SMALL_MATRIX_OPT), 1)
 CCOMMON_OPT += -DSMALL_MATRIX_OPT

diff --git a/cmake/system.cmake b/cmake/system.cmake
@@ -391,7 +391,7 @@ endif ()
 if (X86_64 OR ${CORE} STREQUAL POWER10)
   set(SMALL_MATRIX_OPT TRUE)
 endif ()
-if (ARM64)
+if (ARM64 OR RISCV64 OR POWER)
   set(GEMM_GEMV_FORWARD TRUE)
 endif ()