What problem does it solve?
Optimizes existing Triton kernels for NVIDIA's TileIR backend on Blackwell GPUs by automatically generating and applying TileIR-specific autotune configurations, enabling higher performance through TMA descriptor usage, 2CTA configurations, and occupancy tuning. This skill coordinates the end-to-end workflow from kernel classification to optimization and benchmarking, reducing manual tuning effort and ensuring TileIR compatibility.
Core Features & Use Cases
- TileIR-aware autotuning for dot-related kernels (GEMM, attention) using TMA descriptors and 2CTA strategies.
- Norm-like and reduction kernel support with high-occupancy configurations and varied num_warps.
- Element-wise kernels optimization with extended block sizes and occupancy ranges, plus robust validation against PTX baseline.
- End-to-end workflow: classify kernel, apply TileIR optimizations, validate, and benchmark TileIR vs PTX backends on Blackwell hardware.
Quick Start
Run the classification and TileIR optimization workflow on a sample Triton kernel to generate optimized configs.