arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2103.08825cs.DC
收藏

一种有效的空间矢量化方法

An Efficient Vectorization Scheme for Stencil Computation

Kun Li, Liang Yuan, Yunquan Zhang, Yue Yue, Hang Cao, Pengqi Lu

摘要:

摘要空间计算是各种科学和工程应用中最重要的核心之一。各种工作都集中在矢量化和平铺技术,旨在利用内核内的数据并行和数据局部性。在本文中,现有的矢量化方案的缺点进行了分析。简而言之,它们要么引发数据对齐冲突,要么在与切片集成时损害数据局部性。然后,我们提出了一种新的转置布局,以保持数据的局部性平铺和减少数据重组的矢量化开销的同时。为了进一步提高寄存器级的数据重用,设计了一种时间循环展开和阻塞策略,沿时间维执行多步模板计算。在AVX-2和AVX-512 CPU上的实验结果表明,该方法具有较好的性能。

英文摘要:

Stencil computation is one of the most important kernels in various scientific and engineering applications. A variety of work has focused on vectorization and tiling techniques, aiming at exploiting the in-core data parallelism and data locality respectively. In this paper, the downsides of existing vectorization schemes are analyzed. Briefly, they either incur data alignment conflicts or hurt the data locality when integrated with tiling. Then we propose a novel transpose layout to preserve the data locality for tiling and reduce the data reorganization overhead for vectorization simultaneously. To further improve the data reuse at the register level, a time loop unroll-and-jam strategy is designed to perform multistep stencil computation along the time dimension. Experimental results on the AVX-2 and AVX-512 CPUs show that our approach obtains a competitive performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑