floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #4831 from martin-frbg/gemmforward 

Enable forwarding from GEMM to GEMV for RISCV and PPC in addition to ARM64
This commit is contained in:
Martin Kroeker 2024-08-03 18:52:11 +02:00 committed by GitHub
parent 9afd0c8afd 42d8865234
commit 2c2b6bcf96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Makefile.system
View File

@ -277,6 +277,12 @@ endif
ifeq ($(ARCH), arm64) ifeq ($(ARCH), arm64)
GEMM_GEMV_FORWARD = 1 GEMM_GEMV_FORWARD = 1
endif endif
ifeq ($(ARCH), riscv)
GEMM_GEMV_FORWARD = 1
endif
ifeq ($(ARCH), power)
GEMM_GEMV_FORWARD = 1
endif
ifeq ($(SMALL_MATRIX_OPT), 1) ifeq ($(SMALL_MATRIX_OPT), 1)
CCOMMON_OPT += -DSMALL_MATRIX_OPT CCOMMON_OPT += -DSMALL_MATRIX_OPT

2
cmake/system.cmake
View File

@ -391,7 +391,7 @@ endif ()
if (X86_64 OR ${CORE} STREQUAL POWER10) if (X86_64 OR ${CORE} STREQUAL POWER10)
set(SMALL_MATRIX_OPT TRUE) set(SMALL_MATRIX_OPT TRUE)
endif () endif ()
if (ARM64) if (ARM64 OR RISCV64 OR POWER)
set(GEMM_GEMV_FORWARD TRUE) set(GEMM_GEMV_FORWARD TRUE)
endif () endif ()