// Auto-generated file. Do not edit! // Template: src/f32-rdsum/wasm-simd.c.in // Generator: tools/xngen // // Copyright 2024 Google LLC // // This source code is licensed under the BSD-style license found in the // LICENSE file in the root directory of this source tree. #include #include #include "xnnpack/common.h" #include "xnnpack/reduce.h" #include "xnnpack/math.h" void xnn_f32_rdsum_ukernel_7p7x__wasmsimd_c64( size_t rows, size_t channels, const float* input, size_t input_stride, const float* zero, float* output, const struct xnn_f32_scale_params params[restrict XNN_MIN_ELEMENTS(1)]) { assert(rows != 0); assert(channels != 0); assert(input != NULL); assert(output != NULL); const v128_t vscale = wasm_v128_load32_splat(¶ms->scalar.scale); size_t input_increment = 7 * input_stride; for (; channels >= 64; channels -= 64) { const float* i0 = input; const float* i1 = (const float*) ((uintptr_t) input + 1 * input_stride); const float* i2 = (const float*) ((uintptr_t) input + 2 * input_stride); const float* i3 = (const float*) ((uintptr_t) input + 3 * input_stride); const float* i4 = (const float*) ((uintptr_t) input + 4 * input_stride); const float* i5 = (const float*) ((uintptr_t) input + 5 * input_stride); const float* i6 = (const float*) ((uintptr_t) input + 6 * input_stride); v128_t vacc0 = wasm_i32x4_const_splat(0.f); v128_t vacc1 = wasm_i32x4_const_splat(0.f); v128_t vacc2 = wasm_i32x4_const_splat(0.f); v128_t vacc3 = wasm_i32x4_const_splat(0.f); v128_t vacc4 = wasm_i32x4_const_splat(0.f); v128_t vacc5 = wasm_i32x4_const_splat(0.f); v128_t vacc6 = wasm_i32x4_const_splat(0.f); v128_t vacc7 = wasm_i32x4_const_splat(0.f); v128_t vacc8 = wasm_i32x4_const_splat(0.f); v128_t vacc9 = wasm_i32x4_const_splat(0.f); v128_t vacc10 = wasm_i32x4_const_splat(0.f); v128_t vacc11 = wasm_i32x4_const_splat(0.f); v128_t vacc12 = wasm_i32x4_const_splat(0.f); v128_t vacc13 = wasm_i32x4_const_splat(0.f); v128_t vacc14 = wasm_i32x4_const_splat(0.f); v128_t vacc15 = wasm_i32x4_const_splat(0.f); for (int r = rows; r > 0; r -= 7) { if XNN_UNPREDICTABLE(r < 2) { i1 = zero; } if XNN_UNPREDICTABLE(r <= 2) { i2 = zero; } if XNN_UNPREDICTABLE(r < 4) { i3 = zero; } if XNN_UNPREDICTABLE(r <= 4) { i4 = zero; } if XNN_UNPREDICTABLE(r < 6) { i5 = zero; } if XNN_UNPREDICTABLE(r <= 6) { i6 = zero; } v128_t vin0; v128_t vin1; v128_t vin2; v128_t vin3; v128_t vin4; v128_t vin5; v128_t vin6; v128_t vin7; v128_t vin8; v128_t vin9; v128_t vin10; v128_t vin11; v128_t vin12; v128_t vin13; v128_t vin14; v128_t vin15; vin0 = wasm_v128_load(&i0[0]); vin1 = wasm_v128_load(&i0[4]); vin2 = wasm_v128_load(&i0[8]); vin3 = wasm_v128_load(&i0[12]); vin4 = wasm_v128_load(&i0[16]); vin5 = wasm_v128_load(&i0[20]); vin6 = wasm_v128_load(&i0[24]); vin7 = wasm_v128_load(&i0[28]); vin8 = wasm_v128_load(&i0[32]); vin9 = wasm_v128_load(&i0[36]); vin10 = wasm_v128_load(&i0[40]); vin11 = wasm_v128_load(&i0[44]); vin12 = wasm_v128_load(&i0[48]); vin13 = wasm_v128_load(&i0[52]); vin14 = wasm_v128_load(&i0[56]); vin15 = wasm_v128_load(&i0[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i1[0]); vin1 = wasm_v128_load(&i1[4]); vin2 = wasm_v128_load(&i1[8]); vin3 = wasm_v128_load(&i1[12]); vin4 = wasm_v128_load(&i1[16]); vin5 = wasm_v128_load(&i1[20]); vin6 = wasm_v128_load(&i1[24]); vin7 = wasm_v128_load(&i1[28]); vin8 = wasm_v128_load(&i1[32]); vin9 = wasm_v128_load(&i1[36]); vin10 = wasm_v128_load(&i1[40]); vin11 = wasm_v128_load(&i1[44]); vin12 = wasm_v128_load(&i1[48]); vin13 = wasm_v128_load(&i1[52]); vin14 = wasm_v128_load(&i1[56]); vin15 = wasm_v128_load(&i1[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i2[0]); vin1 = wasm_v128_load(&i2[4]); vin2 = wasm_v128_load(&i2[8]); vin3 = wasm_v128_load(&i2[12]); vin4 = wasm_v128_load(&i2[16]); vin5 = wasm_v128_load(&i2[20]); vin6 = wasm_v128_load(&i2[24]); vin7 = wasm_v128_load(&i2[28]); vin8 = wasm_v128_load(&i2[32]); vin9 = wasm_v128_load(&i2[36]); vin10 = wasm_v128_load(&i2[40]); vin11 = wasm_v128_load(&i2[44]); vin12 = wasm_v128_load(&i2[48]); vin13 = wasm_v128_load(&i2[52]); vin14 = wasm_v128_load(&i2[56]); vin15 = wasm_v128_load(&i2[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i3[0]); vin1 = wasm_v128_load(&i3[4]); vin2 = wasm_v128_load(&i3[8]); vin3 = wasm_v128_load(&i3[12]); vin4 = wasm_v128_load(&i3[16]); vin5 = wasm_v128_load(&i3[20]); vin6 = wasm_v128_load(&i3[24]); vin7 = wasm_v128_load(&i3[28]); vin8 = wasm_v128_load(&i3[32]); vin9 = wasm_v128_load(&i3[36]); vin10 = wasm_v128_load(&i3[40]); vin11 = wasm_v128_load(&i3[44]); vin12 = wasm_v128_load(&i3[48]); vin13 = wasm_v128_load(&i3[52]); vin14 = wasm_v128_load(&i3[56]); vin15 = wasm_v128_load(&i3[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i4[0]); vin1 = wasm_v128_load(&i4[4]); vin2 = wasm_v128_load(&i4[8]); vin3 = wasm_v128_load(&i4[12]); vin4 = wasm_v128_load(&i4[16]); vin5 = wasm_v128_load(&i4[20]); vin6 = wasm_v128_load(&i4[24]); vin7 = wasm_v128_load(&i4[28]); vin8 = wasm_v128_load(&i4[32]); vin9 = wasm_v128_load(&i4[36]); vin10 = wasm_v128_load(&i4[40]); vin11 = wasm_v128_load(&i4[44]); vin12 = wasm_v128_load(&i4[48]); vin13 = wasm_v128_load(&i4[52]); vin14 = wasm_v128_load(&i4[56]); vin15 = wasm_v128_load(&i4[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i5[0]); vin1 = wasm_v128_load(&i5[4]); vin2 = wasm_v128_load(&i5[8]); vin3 = wasm_v128_load(&i5[12]); vin4 = wasm_v128_load(&i5[16]); vin5 = wasm_v128_load(&i5[20]); vin6 = wasm_v128_load(&i5[24]); vin7 = wasm_v128_load(&i5[28]); vin8 = wasm_v128_load(&i5[32]); vin9 = wasm_v128_load(&i5[36]); vin10 = wasm_v128_load(&i5[40]); vin11 = wasm_v128_load(&i5[44]); vin12 = wasm_v128_load(&i5[48]); vin13 = wasm_v128_load(&i5[52]); vin14 = wasm_v128_load(&i5[56]); vin15 = wasm_v128_load(&i5[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); vin0 = wasm_v128_load(&i6[0]); vin1 = wasm_v128_load(&i6[4]); vin2 = wasm_v128_load(&i6[8]); vin3 = wasm_v128_load(&i6[12]); vin4 = wasm_v128_load(&i6[16]); vin5 = wasm_v128_load(&i6[20]); vin6 = wasm_v128_load(&i6[24]); vin7 = wasm_v128_load(&i6[28]); vin8 = wasm_v128_load(&i6[32]); vin9 = wasm_v128_load(&i6[36]); vin10 = wasm_v128_load(&i6[40]); vin11 = wasm_v128_load(&i6[44]); vin12 = wasm_v128_load(&i6[48]); vin13 = wasm_v128_load(&i6[52]); vin14 = wasm_v128_load(&i6[56]); vin15 = wasm_v128_load(&i6[60]); vacc0 = wasm_f32x4_add(vin0, vacc0); vacc1 = wasm_f32x4_add(vin1, vacc1); vacc2 = wasm_f32x4_add(vin2, vacc2); vacc3 = wasm_f32x4_add(vin3, vacc3); vacc4 = wasm_f32x4_add(vin4, vacc4); vacc5 = wasm_f32x4_add(vin5, vacc5); vacc6 = wasm_f32x4_add(vin6, vacc6); vacc7 = wasm_f32x4_add(vin7, vacc7); vacc8 = wasm_f32x4_add(vin8, vacc8); vacc9 = wasm_f32x4_add(vin9, vacc9); vacc10 = wasm_f32x4_add(vin10, vacc10); vacc11 = wasm_f32x4_add(vin11, vacc11); vacc12 = wasm_f32x4_add(vin12, vacc12); vacc13 = wasm_f32x4_add(vin13, vacc13); vacc14 = wasm_f32x4_add(vin14, vacc14); vacc15 = wasm_f32x4_add(vin15, vacc15); i0 = (const float*) ((uintptr_t) i0 + input_increment); i1 = (const float*) ((uintptr_t) i1 + input_increment); i2 = (const float*) ((uintptr_t) i2 + input_increment); i3 = (const float*) ((uintptr_t) i3 + input_increment); i4 = (const float*) ((uintptr_t) i4 + input_increment); i5 = (const float*) ((uintptr_t) i5 + input_increment); i6 = (const float*) ((uintptr_t) i6 + input_increment); } vacc0 = wasm_f32x4_mul(vacc0, vscale); vacc1 = wasm_f32x4_mul(vacc1, vscale); vacc2 = wasm_f32x4_mul(vacc2, vscale); vacc3 = wasm_f32x4_mul(vacc3, vscale); vacc4 = wasm_f32x4_mul(vacc4, vscale); vacc5 = wasm_f32x4_mul(vacc5, vscale); vacc6 = wasm_f32x4_mul(vacc6, vscale); vacc7 = wasm_f32x4_mul(vacc7, vscale); vacc8 = wasm_f32x4_mul(vacc8, vscale); vacc9 = wasm_f32x4_mul(vacc9, vscale); vacc10 = wasm_f32x4_mul(vacc10, vscale); vacc11 = wasm_f32x4_mul(vacc11, vscale); vacc12 = wasm_f32x4_mul(vacc12, vscale); vacc13 = wasm_f32x4_mul(vacc13, vscale); vacc14 = wasm_f32x4_mul(vacc14, vscale); vacc15 = wasm_f32x4_mul(vacc15, vscale); const float* o = output; v128_t vo0 = wasm_v128_load(o); o += 4; v128_t vo1 = wasm_v128_load(o); o += 4; v128_t vo2 = wasm_v128_load(o); o += 4; v128_t vo3 = wasm_v128_load(o); o += 4; v128_t vo4 = wasm_v128_load(o); o += 4; v128_t vo5 = wasm_v128_load(o); o += 4; v128_t vo6 = wasm_v128_load(o); o += 4; v128_t vo7 = wasm_v128_load(o); o += 4; v128_t vo8 = wasm_v128_load(o); o += 4; v128_t vo9 = wasm_v128_load(o); o += 4; v128_t vo10 = wasm_v128_load(o); o += 4; v128_t vo11 = wasm_v128_load(o); o += 4; v128_t vo12 = wasm_v128_load(o); o += 4; v128_t vo13 = wasm_v128_load(o); o += 4; v128_t vo14 = wasm_v128_load(o); o += 4; v128_t vo15 = wasm_v128_load(o); o += 4; vacc0 = wasm_f32x4_add(vo0, vacc0); vacc1 = wasm_f32x4_add(vo1, vacc1); vacc2 = wasm_f32x4_add(vo2, vacc2); vacc3 = wasm_f32x4_add(vo3, vacc3); vacc4 = wasm_f32x4_add(vo4, vacc4); vacc5 = wasm_f32x4_add(vo5, vacc5); vacc6 = wasm_f32x4_add(vo6, vacc6); vacc7 = wasm_f32x4_add(vo7, vacc7); vacc8 = wasm_f32x4_add(vo8, vacc8); vacc9 = wasm_f32x4_add(vo9, vacc9); vacc10 = wasm_f32x4_add(vo10, vacc10); vacc11 = wasm_f32x4_add(vo11, vacc11); vacc12 = wasm_f32x4_add(vo12, vacc12); vacc13 = wasm_f32x4_add(vo13, vacc13); vacc14 = wasm_f32x4_add(vo14, vacc14); vacc15 = wasm_f32x4_add(vo15, vacc15); wasm_v128_store(output, vacc0); output += 4; wasm_v128_store(output, vacc1); output += 4; wasm_v128_store(output, vacc2); output += 4; wasm_v128_store(output, vacc3); output += 4; wasm_v128_store(output, vacc4); output += 4; wasm_v128_store(output, vacc5); output += 4; wasm_v128_store(output, vacc6); output += 4; wasm_v128_store(output, vacc7); output += 4; wasm_v128_store(output, vacc8); output += 4; wasm_v128_store(output, vacc9); output += 4; wasm_v128_store(output, vacc10); output += 4; wasm_v128_store(output, vacc11); output += 4; wasm_v128_store(output, vacc12); output += 4; wasm_v128_store(output, vacc13); output += 4; wasm_v128_store(output, vacc14); output += 4; wasm_v128_store(output, vacc15); output += 4; input = (const float*) ((uintptr_t) input + 64 * sizeof(float)); } if (channels != 0) { input_increment = 7 * input_stride; const float* i0 = input; const float* i1 = (const float*) ((uintptr_t) input + 1 * input_stride); const float* i2 = (const float*) ((uintptr_t) input + 2 * input_stride); const float* i3 = (const float*) ((uintptr_t) input + 3 * input_stride); const float* i4 = (const float*) ((uintptr_t) input + 4 * input_stride); const float* i5 = (const float*) ((uintptr_t) input + 5 * input_stride); const float* i6 = (const float*) ((uintptr_t) input + 6 * input_stride); v128_t vacc[16]; vacc[0] = wasm_i32x4_const_splat(0.f); vacc[1] = wasm_i32x4_const_splat(0.f); vacc[2] = wasm_i32x4_const_splat(0.f); vacc[3] = wasm_i32x4_const_splat(0.f); vacc[4] = wasm_i32x4_const_splat(0.f); vacc[5] = wasm_i32x4_const_splat(0.f); vacc[6] = wasm_i32x4_const_splat(0.f); vacc[7] = wasm_i32x4_const_splat(0.f); vacc[8] = wasm_i32x4_const_splat(0.f); vacc[9] = wasm_i32x4_const_splat(0.f); vacc[10] = wasm_i32x4_const_splat(0.f); vacc[11] = wasm_i32x4_const_splat(0.f); vacc[12] = wasm_i32x4_const_splat(0.f); vacc[13] = wasm_i32x4_const_splat(0.f); vacc[14] = wasm_i32x4_const_splat(0.f); vacc[15] = wasm_i32x4_const_splat(0.f); const size_t num_chunks = round_up_po2(channels, 4) >> 2; for (int r = rows; r > 0; r -= 7) { if XNN_UNPREDICTABLE(r < 2) { i1 = zero; } if XNN_UNPREDICTABLE(r <= 2) { i2 = zero; } if XNN_UNPREDICTABLE(r < 4) { i3 = zero; } if XNN_UNPREDICTABLE(r <= 4) { i4 = zero; } if XNN_UNPREDICTABLE(r < 6) { i5 = zero; } if XNN_UNPREDICTABLE(r <= 6) { i6 = zero; } for (int i = 0; i < num_chunks; ++i) { vacc[i] = wasm_f32x4_add(wasm_v128_load(&i0[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i1[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i2[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i3[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i4[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i5[i*4]), vacc[i]); vacc[i] = wasm_f32x4_add(wasm_v128_load(&i6[i*4]), vacc[i]); } i0 = (const float*) ((uintptr_t) i0 + input_increment); i1 = (const float*) ((uintptr_t) i1 + input_increment); i2 = (const float*) ((uintptr_t) i2 + input_increment); i3 = (const float*) ((uintptr_t) i3 + input_increment); i4 = (const float*) ((uintptr_t) i4 + input_increment); i5 = (const float*) ((uintptr_t) i5 + input_increment); i6 = (const float*) ((uintptr_t) i6 + input_increment); } for (int i = 0; i < num_chunks; ++i) { vacc[i] = wasm_f32x4_mul(vacc[i], vscale); } v128_t vo[16]; const float* o = output; for (int i = 0; i < channels >> 2; ++i) { vo[i] = wasm_v128_load(o); o += 4; } for (int i = 0; i < channels >> 2; ++i) { vacc[i] = wasm_f32x4_add(vo[i], vacc[i]); } for (int i = 0; i < channels >> 2; ++i) { wasm_v128_store(output, vacc[i]); output += 4; } const size_t pos = channels / 4; v128_t vout = vacc[pos]; if (channels & 2) { v128_t vo = wasm_f32x4_make(output[0], output[1], 0.f, 0.f); wasm_v128_store64_lane(output, wasm_f32x4_add(vo, vout), 0); vout = wasm_v64x2_shuffle(vout, vout, 1, 1); output += 2; } if (channels & 1) { *output += wasm_f32x4_extract_lane(vout, 0); } } }