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
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
File details
Details for the file alkompy-0.2.0.post3.tar.gz
.
File metadata
- Download URL: alkompy-0.2.0.post3.tar.gz
- Upload date:
- Size: 3.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.2.0 pkginfo/1.6.1 requests/2.24.0 setuptools/50.3.2 requests-toolbelt/0.9.1 tqdm/4.51.0 CPython/3.8.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | ee5a8749acb443e095dac4c640e658a2539ae795a86b444922659a330ea248c1 |
|
MD5 | 1d351578a33433b22993baf4895588b6 |
|
BLAKE2b-256 | b4ef13f04ce1c8f1f2157d85dfddfe7779998f810dae9b4c478f29baeacc4ac9 |