A Python library using the Futhark C backend via CFFI
Project description
futhark-pycffi
Python library using the Futhark C backend via CFFI
Futhark provides several compilers, futhark-opencl
which is a C backend, and futhark-pyopencl
which is a Python backend based on PyOpenCL. However, the host-side code of the Python backend is quite slow, leading to a lot of overhead when small, frequent kernels are used.
A solution to reduce this overhead is to use CFFI to used the C backend from Python, greatly reducing the calling overhead. The OpenCL code is the same, so this is not interesting for long-running kernels.
This library supports both futhark-opencl
and futhark-c
. Futhark arrays are mapped to and from Numpy arrays. Multiple outputs and multi-dimensional arrays are supported. Input arrays should be in contiguous C order.
Usage
Generate a C library, and build a Python binding for it
futhark-opencl --library test.fut
python build_futhark_ffi.py test
Use the Python wrapper
import numpy as np
import _test
from futhark_ffi import Futhark
test = Futhark(_test)
res = test.test3(np.arange(10))
test.from_futhark(res)
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 Distribution
Built Distributions
Hashes for futhark_ffi-0.6.1b1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4fe42d842296ae3fe11eda59b8fc4c1063fcabc6be446dd08945c7dffa2b9b67 |
|
MD5 | ae7842c28a18504598e0122cc29bb8e2 |
|
BLAKE2b-256 | 537973c67c5c755fff1fc7c5741c061d9eaf3007b59274056424588fdcb96951 |