liger-kernel-dev

Automate Triton kernel development for Liger Kernel with a three-stage pipeline.

6.6k|578|Updated Aug 6, 2024
One-click install
npx skills add https://github.com/linkedin/Liger-Kernel --skill liger-kernel-dev
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: liger-kernel-dev
Source: https://github.com/linkedin/Liger-Kernel/tree/main/.claude/skills/liger-kernel-dev
Command: npx skills add https://github.com/linkedin/Liger-Kernel --skill liger-kernel-dev

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This skill accelerates the end-to-end development of high-performance Triton kernels for Liger Kernel, turning PyTorch descriptions into production-ready Triton implementations with tests and benchmarks.

Core Features & Use Cases

  • Three-stage development workflow (Analyze -> Generate -> Validate) with human review between stages
  • Creation of new kernels from PyTorch operations, or modification of existing kernels
  • Automatic generation of kernel wrappers, tests, and benchmarks, plus optional plotting for performance analysis
  • NVIDIA GPU support and alignment with the Liger Kernel architecture

Quick Start

Provide a PyTorch operation description or kernel specification to initiate a Liger Kernel development workflow.

Frequently Asked Questions about liger-kernel-dev

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I convert a PyTorch operation into a Triton kernel for Liger Kernel?

To convert PyTorch operations into Triton kernels, provide the operation description to initiate a three-stage pipeline that analyzes the logic, generates the kernel, and validates it with unit tests and benchmarks.

What is the workflow for developing new Triton kernels using Liger Kernel?

The Triton kernel development workflow uses a three-stage pipeline: Analyze, Generate, and Validate. Human review is required between each stage to ensure the final implementation meets production-ready standards.

Does Liger Kernel Triton development support AMD GPUs or only NVIDIA GPUs?

Liger Kernel Triton development strictly enforces NVIDIA GPUs only. The automated generation pipeline does not support AMD GPUs or other hardware platforms for kernel validation and benchmarking.

What's the best way to generate unit tests and benchmarks for custom Triton kernels?

The best way to generate unit tests and benchmarks for custom Triton kernels is to use an automated generation pipeline that creates module wrappers and optional performance plots alongside the kernel implementation.

Can I update existing Triton kernels in Liger Kernel or only create new ones?

You can update existing Triton kernels in Liger Kernel as well as create new ones. The development pipeline supports modifying current implementations and converting PyTorch operations into new kernels.

Why does my generated Triton kernel need human review between development stages?

Generated Triton kernels require human review between the Analyze, Generate, and Validate stages to ensure quality control. This step verifies the production-ready implementation before finalizing the unit tests and module wrappers.