Browse Source

Add d8 to d15 to clobber lists as the code does not expressly save them

pull/5423/head
Martin Kroeker GitHub 9 months ago
parent
commit
8e50b8d525
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
  1. +4
    -6
      kernel/arm64/sgemm_direct_alpha_beta_arm64_sme1.c

+ 4
- 6
kernel/arm64/sgemm_direct_alpha_beta_arm64_sme1.c View File

@@ -111,7 +111,6 @@ return;
}
__arm_new("za") __arm_locally_streaming
__attribute__((visibility("hidden")))
static void sgemm_direct_alpha_beta_sme1_2VLx2VL(uint64_t m, uint64_t k, uint64_t n, const float* alpha,\
const float *ba, const float *restrict bb, const float* beta,\
float *restrict C) {
@@ -177,11 +176,11 @@ void CNAME (BLASLONG M, BLASLONG N, BLASLONG K, float alpha, float * __restrict
* of reading directly from vector (z) registers.
* */
asm volatile("" : : :"p0", "p1", "p2", "p3", "p4", "p5", "p6", "p7",
"p8", "p9", "p10", "p11", "p12", "p13", "p14", "p15",
"p8", "p9", "p10", "p11", "p12", "p13", "p14", "p15", "d8", "d9", "d10", "d11", "d12", "d13", "d14", "d15",
"z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7",
"z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15",
"z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23",
"z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31");
"z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31","za");
/* Pre-process the left matrix to make it suitable for
matrix sum of outer-product calculation
@@ -190,11 +189,11 @@ void CNAME (BLASLONG M, BLASLONG N, BLASLONG K, float alpha, float * __restrict
SME1_PREPROCESS(M, K, A, A_mod);
asm volatile("" : : :"p0", "p1", "p2", "p3", "p4", "p5", "p6", "p7",
"p8", "p9", "p10", "p11", "p12", "p13", "p14", "p15",
"p8", "p9", "p10", "p11", "p12", "p13", "p14", "p15","d8", "d9", "d10", "d11", "d12", "d13", "d14", "d15",
"z0", "z1", "z2", "z3", "z4", "z5", "z6", "z7",
"z8", "z9", "z10", "z11", "z12", "z13", "z14", "z15",
"z16", "z17", "z18", "z19", "z20", "z21", "z22", "z23",
"z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31");
"z24", "z25", "z26", "z27", "z28", "z29", "z30", "z31", "za");
/* Calculate C = alpha*A*B + beta*C */
@@ -210,4 +209,3 @@ void CNAME (BLASLONG M, BLASLONG N, BLASLONG K, float alpha, float * __restrict
float beta, float * __restrict R, BLASLONG strideR){fprintf(stderr,"empty sgemm_direct_alpha_beta should not be called!!!\n");}
#endif

Loading…
Cancel
Save