|
|
@@ -420,6 +420,15 @@ 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) |
|
|
@@ -428,6 +437,9 @@ endif () |
|
|
|
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) |
|
|
|
set(CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD") |
|
|
|
endif () |
|
|
|
if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS) |
|
|
|
set(CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD") |
|
|
|
endif () |
|
|
|
if (SMALL_MATRIX_OPT) |
|
|
|
set(CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT") |
|
|
|
endif () |
|
|
|