Browse Source
Merge pull request #4831 from martin-frbg/gemmforward
Enable forwarding from GEMM to GEMV for RISCV and PPC in addition to ARM64
tags/v0.3.28^2
Martin Kroeker
GitHub
1 year ago
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with
7 additions and
1 deletions
-
Makefile.system
-
cmake/system.cmake
|
|
|
@@ -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 |
|
|
|
|
|
|
|
@@ -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 () |
|
|
|
|
|
|
|
|
