A collection of utility functions and tools for Triton projects
Project description
Triton Utils
[WIP]
A collection of utility functions and tools for Triton kernel development.
Most code copied from project below:
- BobMcDear/attorch: A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
- FlagOpen/FlagGems: FlagGems is an operator library for large language models implemented in Triton Language.
- lectures/lecture_014/triton_util.py at main · gpu-mode/lectures
- UmerHA/triton_util: Make triton easier
pre-commit
pip install pre-commit
pre-commit install
pre-commit run --show-diff-on-failure --color=always --all-files
Features
Installation
git clone https://github.com/2niuhe/triton-utils.git && cd triton-utils
pip install .
Usage
Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
CI
- action
.github/workflows/code-format-check.yml
- self runner action
.github/workflows/make-build.yml
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 Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file triton_utils-0.2.0.tar.gz.
File metadata
- Download URL: triton_utils-0.2.0.tar.gz
- Upload date:
- Size: 11.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6501194906297d88b269acb33a14bcbdb624ad3ee92fd5af4983cff7309224ec
|
|
| MD5 |
c0435405b186930838a02ebbcc2d961f
|
|
| BLAKE2b-256 |
c4e9b01ca726816b515e84aa99b6e749a136bca4b5d5396f4c6a294eccc4d8a9
|
File details
Details for the file triton_utils-0.2.0-py3-none-any.whl.
File metadata
- Download URL: triton_utils-0.2.0-py3-none-any.whl
- Upload date:
- Size: 12.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
d7d2fff665a0c1a7a0c89614d1528b7d572779efc4a82fd80dbab7f9c2c46915
|
|
| MD5 |
a40fb97a07bdeef15dd997059b7722d2
|
|
| BLAKE2b-256 |
2ffb6a25b8e683a012214be49a5a418a3e76c030fa0384efc7b00642fd4eec99
|
