python - 与 PyOpenCL 的结构对齐

Question

更新:我的内核中的 int4 是错误的。

我正在使用 pyopencl，但无法使结构对齐正常工作。在下面的代码中，调用内核两次，b 值正确返回(如 1)，但 c 值具有一些“随机”值。

换句话说:我正在尝试读取结构的两个成员。我可以读第一个，但不能读第二个。为什么？

无论我使用 numpy 结构化数组还是使用结构打包，都会出现同样的问题。而且注释中的 _-attribute__ 设置也没有帮助。

我怀疑我在代码的其他地方做了一些愚蠢的事情，但看不到它。任何帮助表示赞赏。

import struct as s
import pyopencl as cl
import numpy as n

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

for use_struct in (True, False):

    if use_struct:
        a = s.pack('=ii',1,2)
        print(a, len(a))
        a_dev = cl.Buffer(ctx, cl.mem_flags.WRITE_ONLY, len(a))
    else:
#       a = n.array([(1,2)], dtype=n.dtype('2i4', align=True))
        a = n.array([(1,2)], dtype=n.dtype('2i4'))
        print(a, a.itemsize, a.nbytes)
        a_dev = cl.Buffer(ctx, cl.mem_flags.WRITE_ONLY, a.nbytes)

    b = n.array([0], dtype='i4')
    print(b, b.itemsize, b.nbytes)
    b_dev = cl.Buffer(ctx, cl.mem_flags.READ_ONLY, b.nbytes)

    c = n.array([0], dtype='i4')
    print(c, c.itemsize, c.nbytes)
    c_dev = cl.Buffer(ctx, cl.mem_flags.READ_ONLY, c.nbytes)

    prg = cl.Program(ctx, """
        typedef struct s {
            int4 f0;
            int4 f1 __attribute__ ((packed));
//            int4 f1 __attribute__ ((aligned (4)));
//            int4 f1;
        } s;
        __kernel void test(__global const s *a, __global int4 *b, __global int4 *c) {
            *b = a->f0;
            *c = a->f1;
        }
        """).build()

    cl.enqueue_copy(queue, a_dev, a)
    event = prg.test(queue, (1,), None, a_dev, b_dev, c_dev)
    event.wait()
    cl.enqueue_copy(queue, b, b_dev)
    print(b)
    cl.enqueue_copy(queue, c, c_dev)
    print(c)

输出(我必须在剪切+粘贴时重新格式化，因此可能会稍微弄乱换行符；我还添加了注释来指示各种打印值是什么):

# first using struct
/home/andrew/projects/personal/kultrung/env/bin/python3.2 /home/andrew/projects/personal/kultrung/src/kultrung/test6.py
b'\x01\x00\x00\x00\x02\x00\x00\x00' 8 # the struct packed values
[0] 4 4                               # output buffer 1
[0] 4 4                               # output buffer 2
/home/andrew/projects/personal/kultrung/env/lib/python3.2/site-packages/pyopencl/cache.py:343: UserWarning: Build succeeded, but resulted in non-empty logs: Build on <pyopencl.Device 'Intel(R) Core(TM)2 CPU         T5600  @ 1.83GHz' at 0x1385a20> succeeded, but said:

Build started Kernel <test> was successfully vectorized Done.   warn("Build succeeded, but resulted in non-empty logs:\n"+message)
[1]         # the first value (correct)
[240]       # the second value (wrong)

# next using numpy
[[1 2]] 4 8 # the numpy struct
[0] 4 4     # output buffer
[0] 4 4     # output buffer
/home/andrew/projects/personal/kultrung/env/lib/python3.2/site-packages/pyopencl/__init__.py:174: UserWarning: Build succeeded, but resulted in non-empty logs: Build on <pyopencl.Device 'Intel(R) Core(TM)2 CPU         T5600  @ 1.83GHz' at 0x1385a20> succeeded, but said:

Build started Kernel <test> was successfully vectorized Done.   warn("Build succeeded, but resulted in non-empty logs:\n"+message)
[1]        # first value (ok)
[67447488] # second value (wrong)

Process finished with exit code 0

Answer 1

在 OpenCL 程序中，尝试结构体本身的 Packed 属性，而不是成员之一:

typedef struct s {
        int4 f0;
        int4 f1;
} __attribute__((packed)) s;

可能是因为您只在结构的单个成员上具有 packed 属性，所以它可能没有打包整个结构。