摘要
本发明公开了一种基于GPU异步复制的SpMV实现方法、系统、装置及介质,属于高性能数值计算技术领域。其中方法包括:S1、采用预设的非零元素划分算法对稀疏矩阵A中的非零元素进行划分,划分得到的每一份元素记为一个batch;S2、将所述batch按照预设的batch分配算法,分配给每个GPU线程块;S3、每个GPU线程块处理各自的batch。本发明使用memcpy-async来实现“batch的加载”,这一新特性能够将线程资源从数据复制任务中释放出来执行其他计算任务,实现了全局内存与共享内存间更快的数据复制,还充分地利用了共享内存的数据复制时间来进行计算,缩短了SpMV整体的时间花销,从而提高了性能。
- 单位