5

以下 python 代码使用 PyOpenCL 用数组 b 中的元素之和填充数组a_plus_b（这不是我的实际目标，但它是我能找到的最简单的代码，但仍然显示了问题）。

import pyopencl as cl
import numpy as np
import numpy.linalg as la

height = 50
width = 32

b = np.arange(width,dtype=np.int32)

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

mf = cl.mem_flags
b_buf = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=b)
dest_buf = cl.Buffer(ctx, mf.WRITE_ONLY, height*4)

prg = cl.Program(ctx, """
    __kernel void sum(__global const int *b, __global int *c)
    {
      int x = get_global_id(1);
      int y;
      c[x] = 0;
      for(y=0;y<get_global_size(0);y++) {
          c[x] += b[y];
      }
    }
    """).build()

prg.sum(queue, (width,height), None, b_buf, dest_buf)

a_plus_b = np.empty(height,dtype=np.int32)
cl.enqueue_copy(queue, a_plus_b, dest_buf)

print(np.sum(b))
print(a_plus_b)
print(np.sum(a_plus_b-np.sum(b)))

给出输出：

496
[496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496
 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496 496
 496 496 496 496 496 496 496 496 496 496 496 496 496 496]
0

但是，如果我将宽度从 32 更改为 33，则数组不再是同一个元素。

528
[555 557 555 559 560 528 560 560 528 528 528 528 528 528 528 528 528 528
 528 528 528 531 540 569 581 528 705 591 560 560 545 560 560 528 560 528
 528 528 528 528 528 528 528 528 528 528 528 532 533 535]
752

事实上，每次运行代码时，都会产生不同的结果。

528
[560 560 559 560 560 560 560 528 528 528 528 528 528 528 528 528 528 528
 528 528 528 560 528 514 565 553 621 650 560 560 560 560 560 528 528 528
 528 528 528 528 528 528 528 528 549 528 528 544 528 537]
724

造成差异的原因是什么？什么不是

4

1 回答 1

2

您正在运行 WIDTH x HEIGHT 工作项。对于内核中的每个 X 值，都会有 WIDTH 工作项并行执行完全相同的操作：将 C[X] 设置为 0，然后在 Y 循环中更新它。所有这些 WIDTH 工作项将读取 C[X]，然后或多或少地同时更新它。这种“或多或少”是您观察到的变化的原因。

您的算法是一维的，您只需要运行 HEIGHT 工作项，并将 WIDTH 作为内核参数传递。将 C[X] 替换为寄存器“SUM”，并在最后执行单个 C[X]=SUM。

于 2013-04-11T21:29:07.617 回答