Skip to content

Commit 82954ba

Browse files
authored
Update ?GEMM-to-?GEMV forwarding settings
1 parent 392d381 commit 82954ba

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

cmake/system.cmake

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -420,6 +420,15 @@ if (X86_64 OR ${CORE} STREQUAL POWER10 OR ARM64 OR LOONGARCH64)
420420
endif ()
421421
if (ARM64)
422422
set(GEMM_GEMV_FORWARD TRUE)
423+
set(SBGEMM_GEMV_FORWARD TRUE)
424+
set(BGEMM_GEMV_FORWARD TRUE)
425+
endif ()
426+
if (POWER)
427+
set(GEMM_GEMV_FORWARD TRUE)
428+
set(SBGEMM_GEMV_FORWARD TRUE)
429+
endif ()
430+
if (RISCV64)
431+
set(GEMM_GEMV_FORWARD TRUE)
423432
endif ()
424433

425434
if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
@@ -428,6 +437,9 @@ endif ()
428437
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
429438
set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD")
430439
endif ()
440+
if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
441+
set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD")
442+
endif ()
431443
if (SMALL_MATRIX_OPT)
432444
set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT")
433445
endif ()

0 commit comments

Comments
 (0)