跳转到主要内容

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]

项目详情


下载文件

下载您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分布

opencl-for-python-0.5.2.tar.gz (497.0 kB 查看哈希值)

上传时间:

由以下支持