cutest

Something cute

Elementwise kernel fuser built on NVIDIA CuTe DSL. Traces a computation graph and generates a single fused CUDA kernel.

(A + B) * C - D fused into one kernel, float16, benchmarked against torch.compile and PyTorch eager on H100.

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
assets		assets
cutest		cutest
rms_norm		rms_norm
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
README.md		README.md
bench_fuse.py		bench_fuse.py
demo_fuse.py		demo_fuse.py

Navigation Menu