Skip to content

Commit 0a53d91

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

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

interface/syrk.c

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -338,12 +338,19 @@ double NNK;
338338
BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME));
339339
return;
340340
}
341+
342+
if (args.n == 0) return;
343+
344+
341345
#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16) && !defined(HFLOAT16)
342346
#if defined(ARCH_ARM64) && (defined(USE_SSYRK_KERNEL_DIRECT)||defined(DYNAMIC_ARCH))
343347
#if defined(DYNAMIC_ARCH)
344-
if (support_sme1())
348+
if (strcmp(gotoblas_corename(), "armv9sme") == 0
349+
#if defined(__clang__)
350+
|| strcmp(gotoblas_corename(), "vortexm4") == 0
351+
#endif
352+
)
345353
#endif
346-
if (args.n == 0) return;
347354
if (order == CblasRowMajor && n == ldc) {
348355
if (Trans == CblasNoTrans && k == lda) {
349356
(Uplo == CblasUpper ? SSYRK_DIRECT_ALPHA_BETA_UN : SSYRK_DIRECT_ALPHA_BETA_LN)(n, k, alpha, a, lda, beta, c, ldc);
@@ -358,7 +365,6 @@ double NNK;
358365

359366
#endif
360367

361-
if (args.n == 0) return;
362368

363369
IDEBUG_START;
364370

0 commit comments

Comments
 (0)