Minor refactoring of RADDSTOREEXPMINUSMAX interface
- Avoid passing raw values in parameters
PiperOrigin-RevId: 419935480
diff --git a/src/f32-raddstoreexpminusmax/neon-p5.c.in b/src/f32-raddstoreexpminusmax/neon-p5.c.in
index e65581b..ee4e006 100644
--- a/src/f32-raddstoreexpminusmax/neon-p5.c.in
+++ b/src/f32-raddstoreexpminusmax/neon-p5.c.in
@@ -19,9 +19,9 @@
void xnn_f32_raddstoreexpminusmax_ukernel__${"neonfma" if FMA else "neon"}_p5_x${ELEMENTS_TILE}${"" if ACCUMULATORS == 1 else "_acc%d" % ACCUMULATORS}(
size_t elements,
const float* input,
+ const float* max,
float* output,
- float* sum,
- float max) XNN_OOB_READS
+ float* sum) XNN_OOB_READS
{
assert(elements % sizeof(float) == 0);
@@ -43,7 +43,7 @@
const float32x4_t vc4 = vmovq_n_f32(0x1.573A1Ap-5f);
const float32x4_t vc5 = vmovq_n_f32(0x1.0F9F9Cp-7f);
- const float32x4_t vi_max = vdupq_n_f32(max);
+ const float32x4_t vi_max = vld1q_dup_f32(max);
$if ELEMENTS_TILE > 4:
$for K in range(ACCUMULATORS):