An extension for Numba to add data-parallel offload capability
Project description
Data-parallel Extension for Numba* (numba-dpex) is a standalone extension for
the Numba Python JIT compiler. Numba-dpex provides
a generic kernel programming API and an offload feature that extends Numba's
auto-parallelizer to generate data-parallel kernels for parfor
nodes.
Numba-dpex's kernel API has a design and API similar to Numba's cuda.jit
module, but is based on the SYCL language. The
code-generation for the kernel API currently supports
SPIR-V-based
OpenCL and
oneAPI Level Zero
devices that are supported by Intel® DPC++ SYCL compiler runtime. Supported
devices include Intel® CPUs, integrated GPUs and discrete GPUs.
The offload functionality in numba-dpex is based on Numba's parfor
loop-parallelizer. Our compiler extends Numba's parfor
feature to generate
kernels and offload them to devices supported by DPC++ SYCL compiler runtime.
The offload functionality is supported via a new NumPy drop-in replacement
library: dpnp. Note that dpnp
and NumPy-based
expressions can be used together in the same function, with dpnp
expressions getting
offloaded by numba-dpex
and NumPy expressions getting parallelized by Numba.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for numba_dpex-0.22.1-1-cp310-cp310-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c3cad11e8b415f4cb83fd52aa45ab59ac20d07a5d1f4394c9e5a9ab7ecf83e02 |
|
MD5 | 6381d45e2f99dc1991ea1c5a2537fbad |
|
BLAKE2b-256 | cbced107d56dc9ccdf352e6f757a4e57d366511022803025498190ae1a96e7d7 |
Hashes for numba_dpex-0.22.1-1-cp310-cp310-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bf117db8f6af780072502dc0f2d2f52bb8d87d37554ba1df95b325e20f74c470 |
|
MD5 | 336ba4aef98f34cb2856b8a7d1ab059e |
|
BLAKE2b-256 | 1591e9f8c02fbe6b3effe2b1a34dbbf1f2d65e79d7ba1abb72f51e4bfde198ed |
Hashes for numba_dpex-0.22.1-1-cp39-cp39-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 18c8817ad6729aa83bf1c1978b68d9d71bd432520a3502da39e6528c3de9db5d |
|
MD5 | 3c8af923868002668819d7db5ca0cfdf |
|
BLAKE2b-256 | 0110662b5e5bf6f5e716e98736a0633a9fa102ed0c6e5ee3c70224ea2d1b486e |
Hashes for numba_dpex-0.22.1-1-cp39-cp39-manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c2cdd42b1633d44d4259b48c67c708186bf3740e3cc997876c2836a27bd34f5b |
|
MD5 | d24315748ab2316a25fecdaf06fe245c |
|
BLAKE2b-256 | 3e26f6dfcc5583b89c79e0fcf60493ce156f55e00419bfdd6c0a8dc2204b9e9e |