File tree Expand file tree Collapse file tree 1 file changed +14
-2
lines changed
Expand file tree Collapse file tree 1 file changed +14
-2
lines changed Original file line number Diff line number Diff line change @@ -420,14 +420,26 @@ 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
425- if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
434+ if (GEMM_GEMV_FORWARD)
426435 set (CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD" )
427436endif ()
428- if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
437+ if (SBGEMM_GEMV_FORWARD)
429438 set (CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD" )
430439endif ()
440+ if (BGEMM_GEMV_FORWARD)
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