diff --git a/cmake/system.cmake b/cmake/system.cmake index 81f1a67ad..544bd192e 100644 --- a/cmake/system.cmake +++ b/cmake/system.cmake @@ -420,14 +420,26 @@ 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) +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) + set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD") +endif () if (SMALL_MATRIX_OPT) set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") endif ()