Browse Source

Update ?GEMM-to-?GEMV forwarding settings

pull/5398/head
Martin Kroeker GitHub 2 months ago
parent
commit
82954ba4ca
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 0 deletions
  1. +12
    -0
      cmake/system.cmake

+ 12
- 0
cmake/system.cmake View File

@@ -420,6 +420,15 @@ if (X86_64 OR ${CORE} STREQUAL POWER10 OR ARM64 OR LOONGARCH64)
endif () endif ()
if (ARM64) if (ARM64)
set(GEMM_GEMV_FORWARD TRUE) 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 () endif ()


if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
@@ -428,6 +437,9 @@ endif ()
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD") set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD")
endif () endif ()
if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD")
endif ()
if (SMALL_MATRIX_OPT) if (SMALL_MATRIX_OPT)
set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT")
endif () endif ()


Loading…
Cancel
Save