作者热门文章
- html - 出于某种原因,IE8 对我的 Sass 文件中继承的 html5 CSS 不友好?
- JMeter 在响应断言中使用 span 标签的问题
- html - 在 :hover and :active? 上具有不同效果的 CSS 动画
- html - 相对于居中的 html 内容固定的 CSS 重复背景?
我正在使用 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 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
希望对您有所帮助。
关于python - 如何在pyopencl中使用float2?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/21745934/
我是一名优秀的程序员,十分优秀!