Browse Source

Merge pull request #5398 from martin-frbg/fixup-5394

Update ?GEMM-to-?GEMV forwarding settings for CMake
pull/5401/head
Martin Kroeker GitHub 2 months ago
parent
commit
4272cf8c7f
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 2 deletions
  1. +14
    -2
      cmake/system.cmake

+ 14
- 2
cmake/system.cmake View File

@@ -420,14 +420,26 @@ 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)
set(CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD") set(CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD")
endif () endif ()
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
if (SBGEMM_GEMV_FORWARD)
set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD") set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD")
endif () endif ()
if (BGEMM_GEMV_FORWARD)
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