sailfish.kernel

A Python module to facilitate JIT-compiled CPU-GPU agnostic compute kernels.

Kernel libraries are collections of functions written in C code that can be compiled for CPU execution using a normal C compiler via the CFFI module, or for GPU execution using a CUDA or ROCm compiler via cupy.

Modules

sailfish.kernel.library

Defines a Library utility class to encapulsate CPU/GPU compiled kernels.

sailfish.kernel.parse_api

Analyzes C code intended for JIT-compilation to a kernel library.

sailfish.kernel.system

Functions for querying compute capabilities and configuring JIT builds.