ARM架构下通过NEON指令集实现基本的加减乘除运算

        在学习了基本的NEON指令集用法之后,就可以使用NEON指令集来实现一些简单的函数了,在这里我们实现一下简单的加减乘除运算函数。需要注意的是,NEON指令对不同类型的数据使用了不同的intrinsic函数,所以编写的函数对于不同类型的数据具体有所区别,但是在逻辑上是相同的。对于不同的数据所用到的intrinsic函数可参见官网:

Intrinsics – Arm Developericon-default.png?t=M85Bhttps://developer.arm.com/architectures/instruction-sets/intrinsics/ 接下来的以float64(double)类型的数据为例

ADD函数

void Add_64f(const double* pSrc1, const double* pSrc2, double* pDst, int len){
           float64x2_t va, vb, vc;
           int l = len - len % 2;
           for(int i=0;i<l;i+=2){
                  va = vld1q_f64(&pSrc1[i]);
		          vb = vld1q_f64(&pSrc2[i]);
                  vc = vAddq_f64(va, vb);
                  vst1q_f64(&pDst[i], vc);
          }
          for(int i=l;i<len;i++){
                  pDst[i] = pSrc1[i] + pSrc2[i];
          }
  }

        最简单的加法函数,读取两个double数组并加载到向量里进行加法运算,最后将结果存储到结果数组中。需要注意的是循环递增的数量要根据向量寄存器里的数据个数来定,如float32位的数据在128位的寄存器中可以存4个,那么这时候循环就应该每次按4递增。除此以外要注意在主循环结束后处理余项

Sub函数

void Sub_64f(const double* pSrc1, const double* pSrc2, double* pDst, int len){
           float64x2_t va, vb, vc;
           int l = len - len % 2;
           for(int i=0;i<l;i+=2){
                  va = vld1q_f64(&pSrc1[i]);
		          vb = vld1q_f64(&pSrc2[i]);
                  vc = vSubq_f64(va, vb);
                  vst1q_f64(&pDst[i], vc);
          }
          for(int i=l;i<len;i++){
                  pDst[i] = pSrc1[i] - pSrc2[i];
          }
  }

基本一致

Mul函数

void Mul_64f(const double* pSrc1, const double* pSrc2, double* pDst, int len){
           float64x2_t va, vb, vc;
           int l = len - len % 2;
           for(int i=0;i<l;i+=2){
                  va = vld1q_f64(&pSrc1[i]);
	              vb = vld1q_f64(&pSrc2[i]);
                  vc = vmulq_f64(va, vb);
                  vst1q_f64(&pDst[i], vc);
          }
          for(int i=l;i<len;i++){
                  pDst[i] = pSrc1[i] * pSrc2[i];
          }
  }

Div函数

void Div_64f(const double* pSrc1, const double* pSrc2, double* pDst, int len){
           float64x2_t va, vb, vc;
           int l = len - len % 2;
           for(int i=0;i<l;i+=2){
                  va = vld1q_f64(&pSrc1[i]);
	              vb = vld1q_f64(&pSrc2[i]);
                  vb = vrecpeq_f64(vb);
                  vc = vmulq_f64(va, vb);
                  vst1q_f64(&pDst[i], vc);
          }
          for(int i=l;i<len;i++){
                  pDst[i] = pSrc1[i] / pSrc2[i];
          }
  }

除法需要注意的是,为了精度考虑,要将除数通过recpe取倒数转化为乘法

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值