Write CUDA GEMM Kernel

Community

High-performance CUDA GEMM kernel design.

AuthorKrxGu
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill guides engineers through designing and implementing a correct, performance-aware CUDA GEMM kernel (C = alpha * A * B + beta * C) with decisions about tiling strategy, memory hierarchy usage, tensor core eligibility, and when to defer to cuBLAS or CUTLASS.

Core Features & Use Cases

  • Guided design of a custom GEMM kernel including tiling, shared memory layout, and epilogue fusion decisions.
  • Decision framework for when to use cuBLAS or CUTLASS and how to handle batched or nonstandard layouts.
  • Applicable in constrained hardware scenarios or research contexts where maximum control over memory and compute is required.

Quick Start

Provide the problem specifications (M, N, K; dtypes; layouts; transpositions; target SM) and initiate the kernel design workflow for a CUDA GEMM.

Dependency Matrix

Required Modules

None required

Components

Standard package

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: Write CUDA GEMM Kernel
Download link: https://github.com/KrxGu/kernel-skills/archive/main.zip#write-cuda-gemm-kernel

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.