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)
420
420
endif ()
421
421
if (ARM64 )
422
422
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 )
423
432
endif ()
424
433
425
- if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
434
+ if (GEMM_GEMV_FORWARD )
426
435
set (CCOMMON_OPT "${CCOMMON_OPT} -DGEMM_GEMV_FORWARD" )
427
436
endif ()
428
- if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
437
+ if (SBGEMM_GEMV_FORWARD )
429
438
set (CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD" )
430
439
endif ()
440
+ if (BGEMM_GEMV_FORWARD )
441
+ set (CCOMMON_OPT "${CCOMMON_OPT} -DBGEMM_GEMV_FORWARD" )
442
+ endif ()
431
443
if (SMALL_MATRIX_OPT )
432
444
set (CCOMMON_OPT "${CCOMMON_OPT} -DSMALL_MATRIX_OPT" )
433
445
endif ()
You can’t perform that action at this time.
0 commit comments