Open CL Python绑定
项目描述
这是另一套OpenCL的Python绑定。
功能
Python 2和Python 3兼容性。
支持OpenCL 1.1
- 可发现属性和方法
不再需要 ctx.get_info(pyopencl.context_info.DEVICES),只需做 ctx.devices
与ctypes紧密集成
import opencl as cl from ctypes import c_float ctx = cl.Context() a = cl.empty(ctx, [2, 3], ctype=c_float)
像Python函数一样调用内核,具有默认参数和关键字参数
import opencl as cl from ctypes import c_float, c_int source = '__kernel void foo(__global float*a, int b, float c) ...' ... # Create a program and context foo = program.foo foo.argnames = 'a', 'b', 'c' foo.argtypes = cl.global_memory(c_float, ndim=2), c_int, c_float # Equivalent to def foo(a, b=1, c=2.0): foo.__defaults__ = 1, 2.0 event = foo(queue, a)
内存对象支持索引和切片
mem2 = memobj[:, 1, :-1]