Skip to content

Commit 4272cf8

Browse files
authored
Merge pull request #5398 from martin-frbg/fixup-5394
Update ?GEMM-to-?GEMV forwarding settings for CMake
2 parents a4f4662 + a5b55f6 commit 4272cf8

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

cmake/system.cmake

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -420,14 +420,26 @@ 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

425-
if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
434+
if (GEMM_GEMV_FORWARD)
426435
set(CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD")
427436
endif ()
428-
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
437+
if (SBGEMM_GEMV_FORWARD)
429438
set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD")
430439
endif ()
440+
if (BGEMM_GEMV_FORWARD)
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)