[发明专利]数据处理方法和装置、电子设备及存储介质

权利要求书

1.一种数据处理方法，其特征在于，所述方法包括：

获取用于表示第一矩阵规模的第一参数，以及用于表示第二矩阵规模的第二参数；

根据所述第一参数以及所述第二参数，确定乘法任务的调度方式，所述乘法任务用于指示将所述第一矩阵与所述第二矩阵相乘，所述乘法任务包括多个子乘法任务，每个子乘法任务用于指示将所述第一矩阵的一个子矩阵与所述第二矩阵的一个子矩阵相乘；

按照所述乘法任务的调度方式，在每次调度过程中，将存在资源共享的多个子乘法任务分配至多个并行执行的线程块进行处理，以经过多次调度过程，得到所述乘法任务对应的处理结果。

2.根据权利要求1所述的方法，其特征在于，所述根据所述第一参数以及所述第二参数，确定乘法任务的调度方式之前，所述方法还包括：

获取资源参数，所述资源参数用于表示并行执行所述乘法任务的进程块数量上限；

根据所述资源参数、所述第一参数以及所述第二参数，判断所述乘法任务所需的进程块数量是否超过并行执行所述乘法任务的进程块数量上限；

若否，执行所述根据所述第一参数以及所述第二参数，确定乘法任务的调度方式的步骤。

3.根据权利要求1或2所述的方法，其特征在于，所述乘法任务的调度方式用于指示在每次调度过程中，存在多个子乘法任务所对应的所述第一矩阵的子矩阵和所述第二矩阵的子矩阵各自对应的数据分别被多次访问。

4.根据权利要求1至3中任意一项所述的方法，其特征在于，所述乘法任务的调度方式用于指示在每次调度过程中，所有子乘法任务所对应的所述第一矩阵的子矩阵和所述第二矩阵的子矩阵对应的数据量最小。

5.根据权利要求1-4任一所述的方法，其特征在于，所述将存在资源共享的多个子乘法任务分配至多个并行执行的线程块进行处理包括：

按照目标映射方式，将存在资源共享的多个子乘法任务分配至多个并行执行的线程块进行处理，其中，每个线程块用于处理所述存在资源共享的多个子乘法任务中的一个乘法任务。

6.根据权利要求5所述的方法，其特征在于，所述按照目标映射方式，将存在资源共享的多个子乘法任务对应分配至多个并行执行的线程块进行处理，包括：

在所述多个并行执行的线程块中，根据行优先规则或列优先规则为所述多个并行执行的线程块依次分配所述存在资源共享的多个子乘法任务，以使所述多个并行执行的线程块处理所述多个子乘法任务。

7.根据权利要求6所述的方法，其特征在于，所述行优先规则包括单个线程块执行子乘法任务过程中，所述单个线程块中奇数行与偶数行的线程排序方向为异向，或是所述单个线程块中每行的线程排序方向为同向；

所述列优先规则包括单个线程块执行子乘法任务过程中，所述单个线程块中奇数列与偶数列的线程排序方向为异向，或是所述单个线程块中每列的线程排序方向为同向。

8.一种数据处理装置，其特征在于，包括：

获取模块，用于获取用于表示第一矩阵规模的第一参数，以及用于表示第二矩阵规模的第二参数；

确定模块，用于根据所述第一参数以及所述第二参数，确定乘法任务的调度方式，所述乘法任务用于指示将所述第一矩阵与所述第二矩阵相乘，所述乘法任务包括多个子乘法任务，每个子乘法任务用于指示将所述第一矩阵的一个子矩阵与所述第二矩阵的一个子矩阵相乘；

处理模块，用于按照所述乘法任务的调度方式，在每次调度过程中，将存在资源共享的多个子乘法任务分配至多个并行执行的线程块进行处理，以经过多次调度过程，得到所述乘法任务对应的处理结果。