Skip to main content

A compute library written in Rust with WebGPU

Project description


Python bindings for alkomp, a GPGPU library written in Rust for performing compute operations.

pip3 install alkompy

At this time, the Python interface is designed to specifically work with numpy ndarrays. This means you can quickly send a numpy array to a GPU with data_gpu = device.to_device(my_np_array) and run a computation using device.call(...). to_device returns an object that records the memory location of a GPU buffer, as well shape and type. In order to retrieve the contents of the buffer: device.get(data_gpu). get function returns a numpy in the same shape as my_np_array.

Build from source

git clone https://github.com/RustyBamboo/alkomp && cd alkomp

pip3 install -r requirements-dev.txt

python3 setup.py develop --user

python3 test/test.py

Project details


Download files

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

Source Distribution

alkompy-0.2.0.post3.tar.gz (3.5 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page