Browse Source

always build sgemm_direct kernel on arm64, even if just as dummy

pull/5222/head
Martin Kroeker GitHub 1 year ago
parent
commit
7241c570d7
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      kernel/CMakeLists.txt

+ 3
- 1
kernel/CMakeLists.txt View File

@@ -208,7 +208,7 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
set(USE_TRMM true)
endif ()
set(USE_DIRECT_SGEMM false)
if (X86_64 OR (ARM64 AND (UC_TARGET_CORE MATCHES ARMV9SME)))
if (X86_64 OR ARM64)
set(USE_DIRECT_SGEMM true)
endif()

@@ -225,9 +225,11 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
set (SGEMMDIRECTSMEKERNEL sgemm_direct_sme1.S)
set (SGEMMDIRECTPREKERNEL sgemm_direct_sme1_preprocess.S)
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL}" "" "gemm_direct" false "" "" false SINGLE)
if (HAVE_SME)
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTSMEKERNEL}" "" "gemm_direct_sme1" false "" "" false SINGLE)
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPREKERNEL}" "" "gemm_direct_sme1_preprocess" false "" "" false SINGLE)
endif ()
endif ()
endif()

foreach (float_type SINGLE DOUBLE)


Loading…
Cancel
Save