Skip to content

Commit 31150eb

Browse files
authored
Move early exit up; don't rely on support_sme() for now
1 parent 0a53d91 commit 31150eb

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

interface/syr2k.c

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -345,12 +345,18 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Tr
345345
return;
346346
}
347347

348+
if (args.n == 0) return;
349+
350+
348351
#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16) && !defined(HFLOAT16)
349352
#if defined(ARCH_ARM64) && (defined(USE_SSYR2K_KERNEL_DIRECT)||defined(DYNAMIC_ARCH))
350353
#if defined(DYNAMIC_ARCH)
351-
if (support_sme1())
354+
if (strcmp(gotoblas_corename(), "armv9sme") == 0
355+
#if defined(__clang__)
356+
|| strcmp(gotoblas_corename(), "vortexm4") == 0
357+
#endif
358+
)
352359
#endif
353-
if (args.n == 0) return;
354360
if (order == CblasRowMajor && n == ldc) {
355361
if (Trans == CblasNoTrans && k == lda && k == ldb) {
356362
if (Uplo == CblasUpper) {
@@ -376,7 +382,6 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Tr
376382

377383
#endif
378384

379-
if (args.n == 0) return;
380385

381386
IDEBUG_START;
382387

0 commit comments

Comments
 (0)