File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -420,6 +420,15 @@ if (X86_64 OR ${CORE} STREQUAL POWER10 OR ARM64 OR LOONGARCH64)
420420endif ()
421421if (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 )
423432endif ()
424433
425434if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
@@ -428,6 +437,9 @@ endif ()
428437if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
429438 set (CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD" )
430439endif ()
440+ if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS)
441+ set (CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD" )
442+ endif ()
431443if (SMALL_MATRIX_OPT)
432444 set (CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT" )
433445endif ()
You can’t perform that action at this time.
0 commit comments