Skip to content

Commit 7241c57

Browse files
authored
always build sgemm_direct kernel on arm64, even if just as dummy
1 parent c81c1c8 commit 7241c57

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

kernel/CMakeLists.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
208208
set(USE_TRMM true)
209209
endif ()
210210
set(USE_DIRECT_SGEMM false)
211-
if (X86_64 OR (ARM64 AND (UC_TARGET_CORE MATCHES ARMV9SME)))
211+
if (X86_64 OR ARM64)
212212
set(USE_DIRECT_SGEMM true)
213213
endif()
214214

@@ -225,9 +225,11 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
225225
set (SGEMMDIRECTSMEKERNEL sgemm_direct_sme1.S)
226226
set (SGEMMDIRECTPREKERNEL sgemm_direct_sme1_preprocess.S)
227227
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL}" "" "gemm_direct" false "" "" false SINGLE)
228+
if (HAVE_SME)
228229
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTSMEKERNEL}" "" "gemm_direct_sme1" false "" "" false SINGLE)
229230
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPREKERNEL}" "" "gemm_direct_sme1_preprocess" false "" "" false SINGLE)
230231
endif ()
232+
endif ()
231233
endif()
232234

233235
foreach (float_type SINGLE DOUBLE)

0 commit comments

Comments
 (0)