Next generation GPU API for Python
Project description
wgpu-py
A Python implementation of WebGPU - the next generation GPU API.
Introduction
In short, this is a Python lib wrapping wgpu-native and exposing it with a Pythonic API similar to the WebGPU spec.
The OpenGL API is old and showing it's cracks. New API's like Vulkan, Metal and
DX12 provide a modern way to control the GPU, but these API's are too low-level
for general use. The WebGPU API follows the same concepts, but with a simpler
(higher level) spelling. The Python wgpu library brings the WebGPU API to
Python.
To get an idea of what this API looks like have a look at triangle.py and the other examples.
Status
Note
The wgpu-API has not settled yet, use with care!
- Coverage of the WebGPU spec is complete enough to build e.g. pygfx.
- Test coverage of the API is close to 100%.
- Support for Windows, Linux, and MacOS (Intel and M1).
- Until WebGPU settles as a standard, its specification may change, and with that our API will probably too. Check the changelog when you upgrade!
Installation
pip install wgpu glfw
Linux users should make sure that pip >= 20.3. That should do the trick on most systems. See getting started for details.
Usage
Also see the online documentation and the examples.
The full API is accessable via the main namespace:
import wgpu
To render to the screen you can use a variety of GUI toolkits:
# The auto backend selects either the glfw, qt or jupyter backend
from wgpu.gui.auto import WgpuCanvas, run, call_later
# Visualizations can be embedded as a widget in a Qt application.
# Import PySide6, PyQt6, PySide2 or PyQt5 before running the line below.
# The code will detect and use the library that is imported.
from wgpu.gui.qt import WgpuCanvas
# Visualizations can be embedded as a widget in a wx application.
from wgpu.gui.wx import WgpuCanvas
Some functions in the original wgpu-native API are async. In the Python API,
the default functions are all sync (blocking), making things easy for general use.
Async versions of these functions are available, so wgpu can also work
well with Asyncio or Trio.
License
This code is distributed under the 2-clause BSD license.
Projects using wgpu-py
- pygfx - A python render engine running on wgpu.
- shadertoy - Shadertoy implementation using wgpu-py.
- tinygrad - deep learning framework
- fastplotlib - A fast plotting library
- xdsl - A Python Compiler Design Toolkit (optional wgpu interpreter)
Developers
- Clone the repo.
- Install devtools using
pip install -r dev-requirements.txt(you can replacepipwithpipenvto install to a virtualenv). - Install wgpu-py in editable mode by running
pip install -e ., this will also install runtime dependencies as needed. - Run
python download-wgpu-native.pyto download the upstream wgpu-native binaries.- Or alternatively point the
WGPU_LIB_PATHenvironment variable to a custom build.
- Or alternatively point the
- Use
black .to apply autoformatting. - Use
flake8 .to check for flake errors. - Use
pytest .to run the tests. - Use
pip wheel --no-deps .to build a wheel.
Updating to a later version of WebGPU or wgpu-native
To update to upstream changes, we use a combination of automatic code generation and manual updating. See the codegen utility for more information.
Testing
The test suite is divided into multiple parts:
pytest -v testsruns the core unit tests.pytest -v examplestests the examples.pytest -v wgpu/__pyinstallertests if wgpu is properly supported by pyinstaller.pytest -v codegenlints the generated binding code.
There are two types of tests for examples included:
Type 1: Checking if examples can run
When running the test suite, pytest will run every example in a subprocess, to
see if it can run and exit cleanly. You can opt out of this mechanism by
including the comment # run_example = false in the module.
Type 2: Checking if examples output an image
You can also (independently) opt-in to output testing for examples, by including
the comment # test_example = true in the module. Output testing means the test
suite will attempt to import the canvas instance global from your example, and
call it to see if an image is produced.
To support this type of testing, ensure the following requirements are met:
- The
WgpuCanvasclass is imported from thewgpu.gui.automodule. - The
canvasinstance is exposed as a global in the module. - A rendering callback has been registered with
canvas.request_draw(fn).
Reference screenshots are stored in the examples/screenshots folder, the test
suite will compare the rendered image with the reference.
Note: this step will be skipped when not running on CI. Since images will have subtle differences depending on the system on which they are rendered, that would make the tests unreliable.
For every test that fails on screenshot verification, diffs will be generated
for the rgb and alpha channels and made available in the
examples/screenshots/diffs folder. On CI, the examples/screenshots folder
will be published as a build artifact so you can download and inspect the
differences.
If you want to update the reference screenshot for a given example, you can grab those from the build artifacts as well and commit them to your branch.
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
File details
Details for the file wgpu-0.14.1.tar.gz.
File metadata
- Download URL: wgpu-0.14.1.tar.gz
- Upload date:
- Size: 150.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | b79dad67af6f2eb5804f6b40307e4ad8cdf1425ea94e8bdbf69f98eb984f5988 |
|
| MD5 | 1f7d364fd7c04f55cb08ab355f93c90b |
|
| BLAKE2b-256 | f3674526e651b3103d91166195cc137be6f71573539e6fd37ead2f8adbdf8127 |
File details
Details for the file wgpu-0.14.1-py3-none-win_amd64.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-win_amd64.whl
- Upload date:
- Size: 2.8 MB
- Tags: Python 3, Windows x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 1cd454e4dda5d66154f0214253ddf2cccf8d937ab72900a44d7843770be208c1 |
|
| MD5 | bed0ce697af2118a4413094a44d405c9 |
|
| BLAKE2b-256 | 08fac9a3ad765d4a3270f456f45b6e52ff9596e4124c9dcb695114b1644f758b |
File details
Details for the file wgpu-0.14.1-py3-none-win32.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-win32.whl
- Upload date:
- Size: 2.5 MB
- Tags: Python 3, Windows x86
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 9c2e869bab3fa59fdc1c67cf507588bd44922805da7e5d07a0566c3130658e8e |
|
| MD5 | 22daa87441737b2d376e30fe89b47bd1 |
|
| BLAKE2b-256 | c539e5aad912dde0b83b63e75e374818f2092fefe5cd429de8487985c9d15f79 |
File details
Details for the file wgpu-0.14.1-py3-none-manylinux_2_28_x86_64.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-manylinux_2_28_x86_64.whl
- Upload date:
- Size: 3.6 MB
- Tags: Python 3, manylinux: glibc 2.28+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 483e8360a479276519f059580c0131ad3dced75b8ad1ccc9f2ef9ae07939c57b |
|
| MD5 | 1b902dc93afa25fa4702f7c73ff11c33 |
|
| BLAKE2b-256 | e6366b10057d3246fc326cac6ba6d04048870769c123985129c5154c0bca2d0c |
File details
Details for the file wgpu-0.14.1-py3-none-manylinux_2_28_aarch64.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-manylinux_2_28_aarch64.whl
- Upload date:
- Size: 3.7 MB
- Tags: Python 3, manylinux: glibc 2.28+ ARM64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | f8a79f3e4a649489302425c9c39e3d50e4faebae558696d7c57ea631dd734acb |
|
| MD5 | abc5dd525b4de2c8646ae987b7b81cde |
|
| BLAKE2b-256 | d16ef5d98b60d5c8f16acaf9f872625dc96cd1977aa41d54e22086ff4d443fb9 |
File details
Details for the file wgpu-0.14.1-py3-none-macosx_11_0_arm64.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-macosx_11_0_arm64.whl
- Upload date:
- Size: 1.9 MB
- Tags: Python 3, macOS 11.0+ ARM64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | b7fff867e82136f39370a2cdfae297db9649ed0da791878e96f27f3d4a33fc26 |
|
| MD5 | 7bc1799534f2872d52ab5a08106d865d |
|
| BLAKE2b-256 | e3b6fe1e736c023445976731c116f47386e9b6446b397903ce6b7e92416b5f70 |
File details
Details for the file wgpu-0.14.1-py3-none-macosx_10_9_x86_64.whl.
File metadata
- Download URL: wgpu-0.14.1-py3-none-macosx_10_9_x86_64.whl
- Upload date:
- Size: 2.0 MB
- Tags: Python 3, macOS 10.9+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | d1b76eb02a937b50ba1c52634507547c9eea62b98907dfbddc974bee905ef2c9 |
|
| MD5 | 9d699db5734a2297c44cc66f1e17e574 |
|
| BLAKE2b-256 | 54364453c25e98fdd55b7bf820f6c302b3085705517bed998ab8fe6b365b843c |
