pyOpenCL和2D FFT

时间:2014-04-02 17:10:49

标签: python opencl fft pyopencl

我正在使用pyFFT对2D数组进行傅里叶变换,然后继续使用另一个OpenCL程序(这里以它为例):

gpu_data = cl_array.to_device(queue, tData2D)  
plan.execute(gpu_data.data)  
eData2D = gpu_data.get()  


ctx = cl.Context([cl.get_platforms()[0].get_devices()[0]])
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags
eData2D_buf = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=eData2D)
eData2D_dest_buf = cl.Buffer(ctx, mf.WRITE_ONLY, eData2D.nbytes)
prg = cl.Program(ctx, """
        //#define PYOPENCL_DEFINE_CDOUBLE     // uncomment for double support.
        #include "pyopencl-complex.h"    
        __kernel void sum(const unsigned int ySize,
                              __global cfloat_t *a,
                              __global cfloat_t *b)
        {
          int gid0 = get_global_id(0);
          int gid1 = get_global_id(1);

          b[gid1 + ySize*gid0] = a[gid1 + ySize*gid0]+a[gid1 + ySize*gid0];
        }
        """).build()

prg.sum(queue, eData2D.shape, None, np.int32(Ny), eData2D_buf, eData2D_dest_buf)
cl.enqueue_copy(queue, eData2Dresult, eData2D_dest_buf) 

这完全没问题。现在,不是检索数据并将其重新复制到缓冲区eData2D = gpu_data.get()中,而是将其重新复制回GPU内存eData2D_buf = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=eData2D),我希望继续使用它。

我期待这样的事情:

gpu_data = cl_array.to_device(queue, tData2D)  
plan.execute(gpu_data.data)


ctx = cl.Context([cl.get_platforms()[0].get_devices()[0]])
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags
eData2D_dest_buf = cl.Buffer(ctx, mf.WRITE_ONLY, eData2D.nbytes)
prg = cl.Program(ctx, """
        //#define PYOPENCL_DEFINE_CDOUBLE     // uncomment for double support.
        #include "pyopencl-complex.h"    
        __kernel void sum(const unsigned int ySize,
                              __global cfloat_t *a,
                              __global cfloat_t *b)
        {
          int gid0 = get_global_id(0);
          int gid1 = get_global_id(1);

          b[gid1 + ySize*gid0] = a[gid1 + ySize*gid0]+a[gid1 + ySize*gid0];
        }
        """).build()

prg.sum(queue, eData2D.shape, None, np.int32(Ny), gpu_data.data, eData2D_dest_buf)
cl.enqueue_copy(queue, eData2Dresult, eData2D_dest_buf) 

哪个不起作用。有没有办法做到这一点? 在此先感谢您的帮助。

1 个答案:

答案 0 :(得分:1)

看起来你正在创建一个全新的上下文来运行第二个程序:

ctx = cl.Context(...)

OpenCL缓冲区仅对创建它的上下文有效,因此如果要重新使用这样的缓冲区,则需要确保对两个OpenCL程序/内核使用相同的上下文。

您也可以重新使用命令队列,而不是创建新的命令队列。