add-cuda-kernel
CommunityGuided CUDA kernel integration for FlashInfer.
Authorsunxxuns
Version1.0.0
Installs0
System Documentation
What problem does it solve?
This Skill provides a practical, end-to-end guide for adding new CUDA kernels to FlashInfer, enabling developers to extend the kernel library with custom operations.
Core Features & Use Cases
- Step-by-step workflow covering kernel implementation, launcher creation, TVM-FFI binding, JIT module generation, Python API exposure, testing, and AOT registration.
- Real-world scenario: integrate a new CUDA kernel into FlashInfer to accelerate a custom op in an LLM serving pipeline.
- Demonstrates how to expose a clean Python API and ensure compatibility with FlashInfer's build and deployment tooling.
Quick Start
Follow the steps in SKILL.md to implement a sample CUDA kernel, including creating include and csrc code, binding, JIT generation, Python API, tests, and AOT registration.
Dependency Matrix
Required Modules
None requiredComponents
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: add-cuda-kernel Download link: https://github.com/sunxxuns/flashinfer/archive/main.zip#add-cuda-kernel Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.