c++ - OpenCL中逐行矩阵运算的实现

Question

我是 OpenCL 的新手，遵循了 OpenCL 中矩阵运算的一些基本教程，但我很难理解如何将 C++ 中的一些循环运算实现到 OpenCL 中。

对于二维数组，matrix A x matrix A 的直接乘法将实现为:

__kernel void sum(
    __global const float *a, __global float *g) {
        const int size = 4;
        int i = get_global_id(1);
        int j = get_global_id(0);

        float z  = a_g[i];
        g [i+size*j] = a[i+size*j] *a[i+size*j]  ;
        }

但是如果我只想将某些行中的元素而不是整个矩阵相乘怎么办？

for (int j=1; j < 4; j++){
    for (int i=0; i < 4; i++){
         P[0][j] += Z[i][0]*Z[i][j];
    }
}

说 P 是一个一维数组，其中 P 的每个元素是每行第一个元素与矩阵 Z 行中所有元素的乘积之和。如何寻址每行中由 Z[i][0] 表示的第一个元素并在 OpenCL 中与 Z[i][j] 相乘？

谢谢

Answer 1

首先，您示例中的内核没有计算标准矩阵乘积。它似乎正在做哈达玛产品。然而，在 CL 中直接翻译 for 循环代码可能如下所示:

__kernel void product(__global const float *Z, __global float *P)
{
        const int size = get_global_size(0);
        int j = get_global_id(0);
        P[j] = 0;
        for (int i = 0; i < size; ++i)
        {
             P[j] += a[size*i] *a[j + size*i];
        }
}

全局大小为 {4}。这不是最佳解决方案，但它是最简单的。