GPU programming with NumbaPro
NumbaPro is a Python compiler that provides a CUDA-based API to write CUDA programs. It is designed for array-oriented computing tasks, much like the widely used NumPy library. The data parallelism in array-oriented computing tasks is a natural fit for accelerators such as GPUs. NumbaPro understands NumPy array types and uses them to generate efficient compiled code for execution on GPUs or multicore CPUs.
The compiler works by allowing you to specify type signatures for Python functions, which enable compilation at runtime (called the JIT compilation).
The most important decorators are:
numbapro.jit: This allows a developer to write CUDA-like functions. When encountered, the compiler translates the code under the decorator ...
Get Python Parallel Programming Cookbook now with the O’Reilly learning platform.
O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.
