File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments