Welcome to PyOpenCL’s documentation!#
PyOpenCL gives you easy, Pythonic access to the OpenCL parallel computation API. What makes PyOpenCL special?
Object cleanup tied to lifetime of objects. This idiom, often called RAII in C++, makes it much easier to write correct, leak- and crash-free code.
Completeness. PyOpenCL puts the full power of OpenCL’s API at your disposal, if you wish. Every obscure get_info() query and all CL calls are accessible.
Automatic Error Checking. All errors are automatically translated into Python exceptions.
Speed. PyOpenCL’s base layer is written in C++, so all the niceties above are virtually free.
Helpful Documentation. You’re looking at it. ;)
Liberal license. PyOpenCL is open-source under the MIT license and free for commercial, academic, and private use.
Here’s an example, to give you an impression:
#!/usr/bin/env python
import numpy as np
import pyopencl as cl
a_np = np.random.rand(50000).astype(np.float32)
b_np = np.random.rand(50000).astype(np.float32)
ctx = cl.create_some_context()
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags
a_g = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=a_np)
b_g = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=b_np)
prg = cl.Program(ctx, """
__kernel void sum(
__global const float *a_g, __global const float *b_g, __global float *res_g)
{
int gid = get_global_id(0);
res_g[gid] = a_g[gid] + b_g[gid];
}
""").build()
res_g = cl.Buffer(ctx, mf.WRITE_ONLY, a_np.nbytes)
knl = prg.sum # Use this Kernel object for repeated calls
knl(queue, a_np.shape, None, a_g, b_g, res_g)
res_np = np.empty_like(a_np)
cl.enqueue_copy(queue, res_np, res_g)
# Check on CPU with Numpy:
print(res_np - (a_np + b_np))
print(np.linalg.norm(res_np - (a_np + b_np)))
assert np.allclose(res_np, a_np + b_np)
(You can find this example as
examples/demo.py
in the PyOpenCL
source distribution.)
Tutorials#
Gaston Hillar’s two-part article series in Dr. Dobb’s Journal provides a friendly introduction to PyOpenCL.
Simon McIntosh-Smith and Tom Deakin’s course Hands-on OpenCL contains both lecture slides and exercises (with solutions) (The course covers PyOpenCL as well as OpenCL’s C and C++ APIs.)
PyOpenCL course at DTU GPULab and Simula (2011): Lecture 1 Lecture 2 Problem set 1 Problem set 2
Ian Johnson’s PyOpenCL tutorial.
Software that works with or enhances PyOpenCL#
Jon Roose’s pyclblas (code) makes BLAS in the form of clBLAS available from within
pyopencl
code.Two earlier wrappers continue to be available: one by Eric Hunsberger and one by Lars Ericson.
Cedric Nugteren provides a wrapper for the CLBlast OpenCL BLAS library: PyCLBlast.
Gregor Thalhammer’s gpyfft provides a Python wrapper for the OpenCL FFT library clFFT from AMD.
Bogdan Opanchuk’s reikna offers a variety of GPU-based algorithms (FFT, random number generation, matrix multiplication) designed to work with
pyopencl.array.Array
objects.Troels Henriksen, Ken Friis Larsen, and Cosmin Oancea’s Futhark programming language offers a nice way to code nested-parallel programs with reductions and scans on data in
pyopencl.array.Array
instances.Robbert Harms and Alard Roebroeck’s MOT offers a variety of GPU-enabled non-linear optimization algorithms and MCMC sampling routines for parallel optimization and sampling of multiple problems.
If you know of a piece of software you feel that should be on this list, please let me know, or, even better, send a patch!
Contents#
- OpenCL Runtime: Basics
- OpenCL Runtime: Constants
- OpenCL Runtime: Platforms, Devices and Contexts
- OpenCL Runtime: Command Queues and Events
- OpenCL Runtime: Memory
- OpenCL Runtime: Programs and Kernels
- OpenCL Runtime: OpenGL Interoperability
- Multi-dimensional arrays
- OpenCL Type Mapping
- Parallel Algorithms
- How-tos
- Built-in Utilities
- Installation
- Tips
- Guidelines
- User-visible Changes
- Version 2020.3
- Version 2020.2
- Version 2018.2
- Version 2018.1
- Version 2017.2
- Version 2017.1
- Version 2016.2
- Version 2016.1
- Version 2015.2.4
- Version 2015.2.3
- Version 2015.2.2
- Version 2015.2.1
- Version 2015.2
- Version 2015.1
- Version 2014.1
- Version 2013.2
- Version 2013.1
- Version 2012.1
- Version 2011.2
- Version 2011.1.2
- Version 2011.1.1
- Version 2011.1
- Version 0.92
- Version 0.91.5
- Version 0.91.4
- Version 0.91.3
- Version 0.91.2
- Version 0.91.1
- Version 0.91
- Version 0.90.4
- Version 0.90.3
- Version 0.90.2
- Version 0.90.1
- Version 0.90
- License
- Frequently Asked Questions
- Citing PyOpenCL
- Acknowledgments
- Documentation Cross-References
- 🚀 Github
- đź’ľ Download Releases
Note that this guide does not explain OpenCL programming and technology. Please refer to the official Khronos OpenCL documentation for that.
PyOpenCL also has its own web site, where you can find updates, new versions, documentation, and support.