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
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 |