Python 如何在pyopencl中使用float2?
我正在使用PyOpenCL编写OpenCL代码。我的内核程序的输入是float2Python 如何在pyopencl中使用float2?,python,opencl,pyopencl,Python,Opencl,Pyopencl,我正在使用PyOpenCL编写OpenCL代码。我的内核程序的输入是float2 __kernel void Pack_Cmplx( __global float2* Data_In, __global float2* Data_Out, int N) 我需要在python中声明一个缓冲区来存储输出并为内核传递输入 python中与float2等效的数据类型是什么?我在numpy上尝试了dtype,但没有成功:(以下是在pyOpenCL程序中使用float2的MWE: import nu
__kernel void Pack_Cmplx( __global float2* Data_In, __global float2* Data_Out, int N)
我需要在python中声明一个缓冲区来存储输出并为内核传递输入
python中与float2等效的数据类型是什么?我在numpy上尝试了dtype,但没有成功:(以下是在
pyOpenCL
程序中使用float2的MWE:
import numpy as np
###################################################
# openCL libraries
###################################################
import pyopencl as cl
import pyopencl.array as cl_array
deviceID = 0
platformID = 0
workGroup=(1,1)
N = 10
testData = np.zeros(N, dtype=cl_array.vec.float2)
dev = cl.get_platforms()[platformID].get_devices()[deviceID]
ctx = cl.Context([dev])
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags
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;
}
""").build()
prg.Pack_Cmplx(queue, (N,1), workGroup, Data_In, np.int32(N))
cl.enqueue_copy(queue, testData, Data_In)
print testData
我希望这会有所帮助。使用
cl\u array.vec.float2
的替代方法是只使用np.float32
类型,并使numpy(和opencl)缓冲区的大小增加一倍
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))
这实际上适用于float2。让我知道是否有更好的方法!dt=np.dtype([((“num1”,np.float16),(“num2,np.float16)])Data_Out=np.empty(100,dtype=dt)”。它应该是两个
np.float32
值的向量,而不是float16。哦,我不知道pyopencl为float2内置了一个数据类型。这很有帮助。