triton-cuda-matmul

What problem does it solve?

This skill automates the generation and optimization of CUDA Triton matmul kernels to maximize performance for large matrix products, enabling efficient GEMM-like workloads.

Core Features & Use Cases

• Tensor Core aware tiling and shared memory strategies for matmul, bmm, and linear layers on NVIDIA GPUs
• Autotune-enabled kernel generation with configurable BLOCK sizes and groupings to suit different M, N, K scales
• Host-side integration patterns for tensor inputs, verification, and benchmarking of generated kernels

Quick Start

Invoke the Triton-CUDA MatMul workflow with your matrix dimensions to generate and autotune a Triton kernel for your MxK and KxN matrices.