手臂霓虹灯指示

时间:2013-03-18 15:17:49

标签: arm vectorization simd neon

我有一些代码,我想使用Neon指令来改变它,但我真的不知道如何完成它...任何人都可以帮助我吗?

void add(int n,float *a,float *b,float t) {
    int i, size = (n+2) * (n+2);
    for(i = 0; i < size; i++)
        a[i] += t * b[i];
}

1 个答案:

答案 0 :(得分:0)

使用NEON内在函数......就像这样:

void add(int n,float *a,float *b,float t) {
  int i, size = (n+2) * (n+2);
  float32x4_t temptt = vdupq_n_f32(t);
  for(i = 0; i < size; i+=4) {
    float32x4_t temp1 = vld1q_f32(a+i);
    float32x4_t temp2 = vld1q_f32(b+i);
    temp1 = vmlaq_f32(temp2, tempt, temp1);
    vst1q_f32(a + i, temp1);
  }
}

这会一次进行4次迭代,并且仅在size为4的倍数时才有效。