Skip to content

Commit 107c883

Browse files
authored
Update SME-related kernels
1 parent 05dbb54 commit 107c883

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

kernel/CMakeLists.txt

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
241241
if (X86_64 OR ARM64)
242242
set(USE_DIRECT_SGEMM true)
243243
endif()
244-
if (UC_TARGET_CORE MATCHES ARMV9SME)
244+
if (UC_TARGET_CORE MATCHES ARMV9SME OR UC_TARGET_CORE MATCHES VORTEXM4)
245245
set (HAVE_SME true)
246246
endif ()
247247

@@ -254,14 +254,16 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
254254
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL}" "" "gemm_direct" false "" "" false SINGLE)
255255
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPERFORMANT}" "" "gemm_direct_performant" false "" "" false SINGLE)
256256
elseif (ARM64)
257+
set (SGEMMDIRECTPERFORMANT sgemm_direct_performant.c)
257258
set (SGEMMDIRECTKERNEL sgemm_direct_arm64_sme1.c)
258259
set (SGEMMDIRECTKERNEL_ALPHA_BETA sgemm_direct_alpha_beta_arm64_sme1.c)
259-
set (SGEMMDIRECTSMEKERNEL sgemm_direct_sme1.S)
260+
set (SGEMMDIRECTSMEKERNEL sgemm_direct_sme1_2VLx2VL.S)
260261
set (SGEMMDIRECTPREKERNEL sgemm_direct_sme1_preprocess.S)
262+
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPERFORMANT}" "" "gemm_direct_performant" false "" "" false SINGLE)
261263
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL}" "" "gemm_direct" false "" "" false SINGLE)
262264
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL_ALPHA_BETA}" "" "gemm_direct_alpha_beta" false "" "" false SINGLE)
263265
if (HAVE_SME)
264-
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTSMEKERNEL}" "" "gemm_direct_sme1" false "" "" false SINGLE)
266+
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTSMEKERNEL}" "" "gemm_direct_sme1_2VLx2VL" false "" "" false SINGLE)
265267
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPREKERNEL}" "" "gemm_direct_sme1_preprocess" false "" "" false SINGLE)
266268
endif ()
267269
endif ()

0 commit comments

Comments
 (0)