From 82954ba4ca43d4c18433445e49af076cf10735d2 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Wed, 23 Jul 2025 23:24:42 +0200 Subject: [PATCH] Update ?GEMM-to-?GEMV forwarding settings --- cmake/system.cmake | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/cmake/system.cmake b/cmake/system.cmake index 81f1a67ad..fa1a4334c 100644 --- a/cmake/system.cmake +++ b/cmake/system.cmake @@ -420,6 +420,15 @@ if (X86_64 OR ${CORE} STREQUAL POWER10 OR ARM64 OR LOONGARCH64) endif () if (ARM64) set(GEMM_GEMV_FORWARD TRUE) + set(SBGEMM_GEMV_FORWARD TRUE) + set(BGEMM_GEMV_FORWARD TRUE) +endif () +if (POWER) + set(GEMM_GEMV_FORWARD TRUE) + set(SBGEMM_GEMV_FORWARD TRUE) +endif () +if (RISCV64) + set(GEMM_GEMV_FORWARD TRUE) endif () if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) @@ -428,6 +437,9 @@ endif () if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD") endif () +if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) + set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD") +endif () if (SMALL_MATRIX_OPT) set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") endif ()