基于SIMD的Square Root函数高性能实现与优化

来源：计算机工程与科学杂志2021年第4期北京时间：

作者：赵永浩, 贾海鹏, 张云泉, 张思佳

单位：1.大连海洋大学信息工程学院，辽宁 大连 116023;2.中国科学院计算技术研究所计算机体系结构国家重点实验室，北京100190

摘要：在计算机图形学、积分计算和神经网络等应用场景中，平方根函数的高性能实现在构建处理器的基础软件生态中起到了十分重要的作用。随着ARM架构处理器得到广泛的使用，研究ARM架构下的函数快速算法实现变得更加关键。当前大量处理器都采用了SIMD架构，所以，研究基于SIMD实现高性能函数计算方法具有重要的研究意义和发展前景。因此，对平方根函数进行了高性能的实现与优化。通过分析IEEE 754标准的浮点数在内存中的存储格式，设计了高效的平方根函数算法；然后通过结合平方根倒数和泰勒公式算法，进一步提高了算法精度；最后通过SIMD优化进一步提升了算法性能。实验结果表明，在满足精度的前提下，相比于libm算法库，实现的平方根函数的，性能提高了约7倍，相比于ARM V8提供的计算平方根的指令在性能上提高了约3倍。

关键词：平方根函数,SIMD,高性能,数值分析,ARM V8架构,

基金资助：国家重点研发计划（2017YFB0202105,2018YFC0809306,2016YFB0200803,2017YFB0202302);国家自然科学基金（61972376);北京市自然科学基金（L182053）