Skills
.Work. You
Rest

add-reference-tests

Official

Automate reference test generation for kernel validation.

Software Engineering#pytest#test-generation#flashinfer#kernel-validation#sglang#reference-implementation
Authorflashinfer-ai
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill automates the creation and validation of pytest-based tests that compare flashinfer_trace reference implementations against FlashInfer or SGLang ground truth, helping maintain kernel correctness as new op_types are added or existing ones are updated.

Core Features & Use Cases

  • Automated test scaffolding: generates tests under flashinfer_trace/tests/references/ for new definitions and op_types.
  • Ground-truth validation: ensures reference implementations match the chosen ground truth (FlashInfer by default, with SGLang as fallback when needed).
  • Reproducible testing: provides deterministic inputs and standardized test templates to support consistent results across environments.

Quick Start

/add-reference-tests --op-type mla_paged /add-reference-tests --op-type moe /add-reference-tests --op-type gqa_paged /add-reference-tests --op-type rmsnorm

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: add-reference-tests
Download link: https://github.com/flashinfer-ai/flashinfer-bench/archive/main.zip#add-reference-tests

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 223,000+ vetted skills library on demand.