Skip to content

Commit e8bd97a

Browse files
authored
add RISCV64 entries for DYNAMIC_ARCH
1 parent f408194 commit e8bd97a

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

cmake/arch.cmake

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,11 @@ if (DYNAMIC_ARCH)
5757
set(DYNAMIC_CORE POWER6 POWER8 POWER9 POWER10)
5858
set(CCOMMON_OPT "${CCOMMON_OPT} -DHAVE_P10_SUPPORT")
5959
endif ()
60-
60+
61+
if (RISCV64)
62+
set(DYNAMIC_CORE RISCV64_GENERIC RISCV64_ZVL128B RISCV64_ZVL256B)
63+
endif ()
64+
6165
if (X86)
6266
set(DYNAMIC_CORE KATMAI COPPERMINE NORTHWOOD PRESCOTT BANIAS CORE2 PENRYN DUNNINGTON NEHALEM ATHLON OPTERON OPTERON_SSE3 BARCELONA BOBCAT ATOM NANO)
6367
endif ()

cmake/system.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ endif ()
391391
if (X86_64 OR ${CORE} STREQUAL POWER10)
392392
set(SMALL_MATRIX_OPT TRUE)
393393
endif ()
394-
if (ARM64 OR RISCV64 OR POWER)
394+
if (ARM64)
395395
set(GEMM_GEMV_FORWARD TRUE)
396396
endif ()
397397

@@ -403,7 +403,7 @@ if (SMALL_MATRIX_OPT)
403403
endif ()
404404

405405
if (DYNAMIC_ARCH)
406-
if (X86 OR X86_64 OR ARM64 OR POWER)
406+
if (X86 OR X86_64 OR ARM64 OR POWER OR RISCV64)
407407
set(CCOMMON_OPT "${CCOMMON_OPT} -DDYNAMIC_ARCH")
408408
if (DYNAMIC_OLDER)
409409
set(CCOMMON_OPT "${CCOMMON_OPT} -DDYNAMIC_OLDER")

0 commit comments

Comments
 (0)