WAVE Python SDK
Write GPU kernels in Python, run on any GPU.
Install
pip install wave-gpu
Usage
import wave_gpu
@wave_gpu.kernel
def vector_add(a: wave_gpu.f32[:], b: wave_gpu.f32[:], out: wave_gpu.f32[:], n: wave_gpu.u32):
gid = wave_gpu.thread_id()
if gid < n:
out[gid] = a[gid] + b[gid]
a = wave_gpu.array([1.0, 2.0, 3.0, 4.0])
b = wave_gpu.array([5.0, 6.0, 7.0, 8.0])
out = wave_gpu.zeros(4)
vector_add(a, b, out, len(a))
print(out.to_list()) # [6.0, 8.0, 10.0, 12.0]
License
Apache 2.0 - see LICENSE
1 maintainer
Metadata
1 maintainer
Release files for wave-gpu 0.1.2
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| wave_gpu-0.1.2.tar.gz | 7.2 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| wave_gpu-0.1.2-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 14.7 kB
Release files / wave_gpu-0.1.2.tar.gz
| Download URL | wave_gpu-0.1.2.tar.gz |
|---|---|
| Size | 7.2 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
e8807f38532b0fa56230905b54de72a72c37de7e083401088d46a6dfd9d44913
|
|
BLAKE2b-256 checksum How to use checksums |
470b0298c304f90716a1e512d22001bf7e08de1d2c82a23a4aa81cf86de05842
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.1.0 CPython/3.13.7
|
Release files / wave_gpu-0.1.2-py3-none-any.whl
| Download URL | wave_gpu-0.1.2-py3-none-any.whl |
|---|---|
| Size | 7.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d39dc7a60713560d8cd3a039ad1890d4e0cb79c77dd09fd6d104771511787956
|
|
BLAKE2b-256 checksum How to use checksums |
e4d6d3d0fe8235a8df8d637a29e4c7f13d5723b88931cc990466b33075745e1f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.1.0 CPython/3.13.7
|
