diff --git a/Makefile.system b/Makefile.system index 436245c35..b065f9a98 100644 --- 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 index efb7aef94..a3f7fd173 100644 --- 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 ()