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)
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
434
if (GEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
@@ -428,6 +437,9 @@ endif ()
428
437
if (SBGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
429
438
set (CCOMMON_OPT "${CCOMMON_OPT} -DSBGEMM_GEMV_FORWARD" )
430
439
endif ()
440
+ if (BGEMM_GEMV_FORWARD AND NOT ONLY_CBLAS )
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