anthony-maio

Triton Fused Epilogue Kernels

Name: Triton Fused Epilogue Kernels
Author: anthony-maio

Development community

SOURCE https://github.com/anthony-maio/triton-skills/blob/main/skills/triton-fused-epilogue-kernels.md

Description

--- name: triton-fused-epilogue-kernels description: Fuse epilogue ops (normalization, gating, residual, activation, dropout) into Triton attention/matmul kernels to eliminate HBM round-trips. --- # Fused Epilogue Kernels in Triton > **Targets:** Triton >= 2.1, SM70+/CDNA2+ ## Overview Fusing epilogue work directly into attention or GEMM kernels avoids extra HBM writes/reads and kernel launches. Perform all final math in-register immediately before the final `tl.store`. Use `tl.constexpr` bo

Installation

Installs to ~/.claude/skills/anthony-maio-triton-skills-triton-fused-epilogue-kernels/SKILL.md

Terminal

mkdir -p ~/.claude/skills/anthony-maio-triton-skills-triton-fused-epilogue-kernels && curl -fsSL https://raw.githubusercontent.com/anthony-maio/triton-skills/HEAD/skills/triton-fused-epilogue-kernels.md -o ~/.claude/skills/anthony-maio-triton-skills-triton-fused-epilogue-kernels/SKILL.md

Restart Claude Code, or start a new session, for it to be picked up.