数据矩阵相乘的加速运算方法和装置

摘要

本发明公开了一种数据矩阵相乘的加速运算方法和装置。其中，该数据矩阵相乘的加速运算方法包括：根据显存容量和待求解矩阵的数据量确定计算待求解矩阵的数据量的单次计算数据量上限；根据数据量上限和待求解矩阵的维度对待求解矩阵进行划分操作，得到多个数据块，其中，每个数据块的数据量小于等于数据量上限；在统一计算设备架构平台上构建线程和线程块的计算结构；在计算结构中依次对多个数据块进行并行运算；以及累计由计算结构依次输出的多个数据块的运算结果。通过本发明，解决了现有技术中进行矩阵相乘的加速运算时引入误差而导致计算结果不准确的问题，达到了提高运算速度的同时保证运算结果的准确的效果。

主权项

一种数据矩阵相乘的加速运算方法，其特征在于，包括：根据显存容量和待求解矩阵的数据量确定计算所述待求解矩阵的数据量的单次计算数据量上限；根据所述数据量上限和所述待求解矩阵的维度对所述待求解矩阵进行划分操作，得到多个数据块，其中，每个所述数据块的数据量小于等于所述数据量上限；在统一计算设备架构平台上构建线程和线程块的计算结构；在所述计算结构中依次对所述多个数据块进行并行运算；以及累计由所述计算结构依次输出的所述多个数据块的运算结果。