RI Kernels

CPU and GPU kernels for radio interferometry, exposed to JAX as FFI custom calls with full support for automatic differentiation.

Installation

The CPU kernels and all of the Python code live in ri_kernels. The GPU kernels ship as add-on packages, one per CUDA major version, selected with an extra:

pip install ri_kernels               # CPU kernels only
pip install "ri_kernels[cuda12]"     # + CUDA 12 kernels and jax[cuda12]
pip install "ri_kernels[cuda13]"     # + CUDA 13 kernels and jax[cuda13]

Building from source

Requires CMake >= 3.20, a C++20 compiler, and network access at configure time (Google Highway is fetched by CMake).

pip install .                      # CPU only
RI_KERNELS_CUDA=1 pip install .    # CPU + CUDA, needs nvcc
RI_KERNELS_ROCM=1 pip install .    # CPU + ROCm, needs hipcc

Both libraries land in the ri_kernels package directory in that case, which is also a location the loader searches. Note that pip install .[cuda12] is a different thing: the extra resolves ri_kernels_cuda12 from the index, so it needs a published release. To build the GPU kernels from the checkout, use RI_KERNELS_CUDA=1.

Other CMake options of note: RI_KERNELS_CPU (default ON), RI_KERNELS_MULTI_ARCH (dynamic SIMD dispatch, default ON — turn it off and set arch flags via CMAKE_CXX_FLAGS for a single-target build), RI_KERNELS_BUNDLED_HIGHWAY, and CMAKE_CUDA_ARCHITECTURES.

Tests

Tests can be run with

python -m pytest tests

Project links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.2.1)

2 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for AdhocMan from gravatar.com AdhocMan Avatar for chrisjfinlay from gravatar.com chrisjfinlay

Credits

Author/Maintainer: Chris Finlay

License

GNU GENERAL PUBLIC LICENSE

Requires

Python >=3.10

Provides Extra

jax

Report project as malware

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl (1.3 MB view details)

Uploaded Python 3manylinux: glibc 2.24+ x86-64manylinux: glibc 2.28+ x86-64

ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl (1.3 MB view details)

Uploaded Python 3manylinux: glibc 2.24+ ARM64manylinux: glibc 2.28+ ARM64

File details

Details for the file ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl.

File metadata

File hashes

Hashes for ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
Algorithm Hash digest
SHA256 ab11931e5efdc2c5220300a156164b74392b970a36d54a12617ade22649b38cc
MD5 b829cec3314aa35b2862cb8a87df8d85
BLAKE2b-256 1a6c603998a5f2c3b27632a878d17b9165e8ca2cbaaec89111998cbb54469233

See more details on using hashes here.

Provenance

The following attestation bundles were made for ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl:

Publisher: wheels.yml on epfl-radio-astro/ri-kernels

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl.

File metadata

File hashes

Hashes for ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl
Algorithm Hash digest
SHA256 a96d45ca71b65f53c218c65bb24a5ec7514c39b0523d0b671aa52fe0411e4cf8
MD5 69756eee2e2c21c507bcc97d8c2aa311
BLAKE2b-256 b5d10d67b45f778f2fde372abc2933584bdedfcc542da8c7270bd8ea9f9e2c5c

See more details on using hashes here.

Provenance

The following attestation bundles were made for ri_kernels_cuda12-0.1.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl:

Publisher: wheels.yml on epfl-radio-astro/ri-kernels

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

0.2.1

2 files

0.2.0

2 files

This release

0.1.0 This release

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page