PyOpenCL 多维数组

Posted

技术标签:

【中文标题】PyOpenCL 多维数组【英文标题】:PyOpenCL Multidimensional Array 【发布时间】:2017-07-25 13:00:22 【问题描述】:

我有这段代码用于使用 PyOpenCL 进行多维数组添加。我的问题是除了第一个维度之外的所有结果都是错误的。我一直在咨询这个Link。

from __future__ import absolute_import, print_function
import numpy as np
import pyopencl as cl

N = 4
a_np = np.random.rand(N,N).astype(np.float32)
b_np = np.random.rand(N,N).astype(np.float32)

ctx = cl.create_some_context()
queue = cl.CommandQueue(ctx)

mf = cl.mem_flags
a_g = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=a_np)
b_g = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=b_np)

prg = cl.Program(ctx, """
    __kernel void sum(
        __global const float *a_g, __global const float *b_g, __global float *res_g) 
            int i = get_global_id(1);
            int j = get_global_id(0);
            res_g[i,j] = a_g[i,j] + b_g[i,j];
    
""").build()

res_g = cl.Buffer(ctx, mf.WRITE_ONLY, a_np.nbytes)
prg.sum(queue, a_np.shape, None, a_g, b_g, res_g)

res_np = np.empty_like(a_np)
cl.enqueue_copy(queue, res_np, res_g)

# Check on CPU with Numpy:
print(res_np - (a_np + b_np))
print(np.linalg.norm(res_np - (a_np + b_np)))
print (res_np)
print (a_np + b_np)

矩阵 1:

[[ 0.2990678   0.76585543  0.71866363  0.30202991]
 [ 0.20604192  0.01989171  0.02402978  0.82826865]
 [ 0.75456071  0.62410605  0.4374246   0.85372066]
 [ 0.37000021  0.5734672   0.4250721   0.2456535 ]]

矩阵 2:

[[ 0.83109927  0.53289926  0.24182947  0.39531609]
 [ 0.53014964  0.62028325  0.2397541   0.03364789]
 [ 0.83543158  0.1162187   0.21168791  0.22438531]
 [ 0.2178313   0.76118374  0.23737679  0.41660839]]

预期结果:

[[ 1.13016701  1.29875469  0.96049309  0.69734597]
 [ 0.73619157  0.64017498  0.26378387  0.86191654]
 [ 1.58999228  0.74032474  0.64911252  1.07810593]
 [ 0.5878315   1.33465099  0.66244888  0.6622619 ]]

脚本结果:

[[ 1.13016701  1.29875469  0.96049309  0.69734597]
 [ 0.          0.          0.          0.        ]
 [ 0.          0.          0.          0.        ]
 [ 0.          0.          0.          0.        ]]

【问题讨论】:

【参考方案1】:

问题出在这里:

res_g[i,j] = a_g[i,j] + b_g[i,j];

这不是您访问OpenCL 中多维数组元素的方式。 OpenCLC 的子集,由***提供:

在 C 和 C++ 编程语言中,逗号运算符 (由记号 , 表示)是一个二元运算符,它计算其 第一个操作数并丢弃结果,然后计算第二个操作数 操作数并返回此值(和类型)。

如此有效地评估为: res_g[j] = a_g[j] + b_g[j];

因此正确的应该是这样的:

res[i + size * j] = ...

再次查看您提供的链接,一切都在那里。

【讨论】:

以上是关于PyOpenCL 多维数组的主要内容,如果未能解决你的问题,请参考以下文章

多维数组索引 C++ 中的多维数组

多维数组指向多维数组

C 语言数组 ( 多维数组本质 | 步长角度 理解 多维数组本质 )

如果数组存在于另一个多维数组中,如何从多维数组中删除该数组? [复制]

php 多个多维数组求交集

php中多维数组的问题