testData = np.zeros(N*2, dtype=np.float32)
Data_In = cl.Buffer(ctx, mf.READ_WRITE, testData.nbytes)
prg = cl.Program(ctx, """
__kernel void Pack_Cmplx( __global float2* Data_In, int N)
{
int gid = get_global_id(0);
Data_In[gid] = 1; // not sure about this tbh. do we set both values to 1 here ?
}
""").build()
prg.Pack_Cmplx(queue, (N,1), workGroup, Data_In, np.int32(N))
使用
cl_array.vec.float2
的另一种方法是只使用np.float32
类型并使numpy(和opencl)缓冲区增大一倍。在{cdmw2>程序中使用{cdmwe}:
我希望这有帮助。在
相关问题 更多 >
编程相关推荐