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] 所示并乘以 Z[i][j]在 OpenCL 中？
谢谢

Answer 1

首先，您示例中的内核不是计算标准矩阵乘积。它似乎在做一个Hadamard产品。但是，在 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}。这不是最佳解决方案，但它是最简单的。