furiosa-kernels 2026.3.0

TCL kernels for NPU

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_kernels 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_kernels 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