ctypes wrappers for HIP, CUDA, and OpenCL
Project description
gpuctypes (a low level GPU library for Python)
Do you wish that there were simple wrappers for GPU libraries in Python? Like not pyopencl, pycuda, and (HIP library someday?) here, but just raw access to the APIs?
That's what gpuctypes does! While high level libraries have their place, the world needs more low level libraries. Like gpuctypes. Welcome home.
Installation (usage)
pip install gpuctypes
Usage
import gpuctypes.hip as hip
import gpuctypes.cuda as cuda
import gpuctypes.opencl as opencl
How it works
gpuctypes uses ctypeslib to autogenerate Python files from the headers of the respective libraries.
Installation (development)
git clone https://github.com/tinygrad/gpuctypes.git
cd gpuctypes
pip install -e .
Current versions
- ROCm 6.0.0
- CUDA 11.5
- OpenCL (whatever is in Ubuntu 22.04)
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
gpuctypes-0.3.0.tar.gz
(91.9 kB
view hashes)
Built Distribution
gpuctypes-0.3.0-py3-none-any.whl
(90.0 kB
view hashes)
Close
Hashes for gpuctypes-0.3.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3510bd3ee779ea79836413b0bd0c0d2943b7616fd17d17c75ea1de903449b85b |
|
MD5 | c9aa3832883e7a26fa24d52464102b12 |
|
BLAKE2b-256 | 65063ed404d0ea3ca525e021d6ea5a94e2173721f7488ca09e28412af238c1bf |