From 82954ba4ca43d4c18433445e49af076cf10735d2 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Wed, 23 Jul 2025 23:24:42 +0200 Subject: [PATCH 1/2] 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 () From a5b55f6fe349ebe032a866ebf8daeb1f1acdc1d8 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Thu, 24 Jul 2025 09:30:58 +0200 Subject: [PATCH 2/2] remove CBLAS restriction on GEMM_GEMV forwarding --- cmake/system.cmake | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cmake/system.cmake b/cmake/system.cmake index fa1a4334c..544bd192e 100644 --- a/cmake/system.cmake +++ b/cmake/system.cmake @@ -431,13 +431,13 @@ if (RISCV64) set(GEMM_GEMV_FORWARD TRUE) endif () -if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) +if (GEMM_GEMV_FORWARD) set(CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD") endif () -if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) +if (SBGEMM_GEMV_FORWARD) set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD") endif () -if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) +if (BGEMM_GEMV_FORWARD) set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD") endif () if (SMALL_MATRIX_OPT)