ARM Neon - 累加点积到输出的索引

Question

我想用下面的代码片段实现向量点积，其中向量a和b的点积可以在第一个位置添加到c，因此结果将是[30, 0, 0, 0]。

#include <arm_neon.h>
#include "utils.h"
int main() {
float a[4]{1, 2, 3, 4};
float b[4]{1, 2, 3, 4};
float c[4];
float32x4_t A = vld1q_f32(a);
float32x4_t B = vld1q_f32(b);
float32x4_t C = vmovq_n_f32(0);
// Computation
vst1q_f32(c, C);
print_vector(c, 4);
}

是否有一个习惯的方式来做这个与Neon?

Answer 1

C = vmulq_f32(A, B);
float32x4_t zero = vmovq_n_f32(0);
vst1q_f32(c, zero);
float32x2_t D = vpadd_f32(vget_low_f32(C), vget_high_f32(C));
D = vpadd_f32(D, D);
vst1_lane_f32(c, D, 0);