Skip to content

Commit 2c2b6bc

Browse files
authored
Merge pull request #4831 from martin-frbg/gemmforward
Enable forwarding from GEMM to GEMV for RISCV and PPC in addition to ARM64
2 parents 9afd0c8 + 42d8865 commit 2c2b6bc

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

Makefile.system

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,6 +277,12 @@ endif
277277
ifeq ($(ARCH), arm64)
278278
GEMM_GEMV_FORWARD = 1
279279
endif
280+
ifeq ($(ARCH), riscv)
281+
GEMM_GEMV_FORWARD = 1
282+
endif
283+
ifeq ($(ARCH), power)
284+
GEMM_GEMV_FORWARD = 1
285+
endif
280286

281287
ifeq ($(SMALL_MATRIX_OPT), 1)
282288
CCOMMON_OPT += -DSMALL_MATRIX_OPT

cmake/system.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ endif ()
391391
if (X86_64 OR ${CORE} STREQUAL POWER10)
392392
set(SMALL_MATRIX_OPT TRUE)
393393
endif ()
394-
if (ARM64)
394+
if (ARM64 OR RISCV64 OR POWER)
395395
set(GEMM_GEMV_FORWARD TRUE)
396396
endif ()
397397

0 commit comments

Comments
 (0)