furiosa-kernels
A Python package providing TCL kernels as reusable Kernel data objects.
Install
cd /path/to/npu-tools/furiosa-kernels
pip install -e ".[dev]"
Test
pytest tests/ -v
Usage
Load a kernel from a .tc file
from furiosa.tcl import Kernel
k = Kernel.from_file(name='my_kernel', path='path/to/my_kernel.tc')
print(k.dsl)
Build a kernel with Python eDSL
import furiosa.tcl as tcl
from furiosa.tcl import Kernel
@tcl.kernel
def my_kernel(
weight: bf16[E, D, F],
gather_index: i32[G],
) -> bf16[G, D, F]:
output: bf16[G, D, F] = tcl.gather(weight, gather_index, E)
return output
k = Kernel.from_edsl(my_kernel)
print(k.dsl)
CLI: convert eDSL to DSL
scripts/run_kernel.py converts @tcl.kernel functions to TCL IR:
# Print generated DSL for a kernel
python scripts/run_kernel.py convert path/to/kernels.py my_kernel
# List all @tcl.kernel functions in a file
python scripts/run_kernel.py list path/to/kernels.py
1 maintainer
Metadata
1 maintainer
Release files for furiosa-kernels 2026.4.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| furiosa_kernels-2026.4.0-py3-none-any.whl | Python 3 | none | any | Details |
Release files / furiosa_kernels-2026.4.0-py3-none-any.whl
| Download URL | furiosa_kernels-2026.4.0-py3-none-any.whl |
|---|---|
| Size | 1.1 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
25df60f7d5c001f8aa015830a29d57888a39edfc77526ffe05a93ab87e0cf72b
|
|
BLAKE2b-256 checksum How to use checksums |
db2fbfbd913eeb69bb97ce086365d42e9497865846bad5cbeb52ed6129c0ce64
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.10.22
|
