使用PyOpenCL进行结构对齐

时间:2011-10-01 15:03:31

标签: python c struct opencl pyopencl

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

我正在使用pyopencl,但无法使struct alignment正常工作。在下面调用内核两次的代码中,b值正确返回(为1),但c值有一些“随机”值。

换句话说: 我试图读取结构的两个成员。我可以读第一个而不是第二个。为什么?

无论是使用numpy结构化数组还是使用struct with pack,都会出现同样的问题。而评论中的_-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

2 个答案:

答案 0 :(得分:0)

在OpenCL程序中,尝试struct本身的packed属性,而不是其中一个成员:

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

可能是因为你在结构的一个成员上只有packed属性,它可能没有打包整个结构。

答案 1 :(得分:0)

好吧,我不知道从哪里得到int4 - 我认为它必须是一个英特尔扩展。使用int切换到AMD,因为内核类型按预期工作。一旦我清理了东西,我会在http://acooke.org/cute/Somesimple0.html发布更多信息。