gpu-kernel-baseline

Installation
SKILL.md

GPU Kernel Baseline

When to Use

Use this skill when the user provides PyTorch logic or a kernel demo and asks to:

  • Write a GPU kernel for the target platform.
  • Build a baseline from scratch.
  • Prepare kernel.py, reference.py, test_kernel.py, and baseline_report.md for later profile-driven optimization.

Workflow

This stage first understands the PyTorch semantics, then learns the framework APIs (CuteDSL or FlyDSL) through <gpu-wiki>/README.md, implements kernel.py and test_kernel.py, validates correctness, records performance, writes baseline_report.md, and writes memory/v0.json.

The orchestrator exposes the knowledge base at ./gpu-wiki/ inside each campaign workspace, referenced below as <gpu-wiki>/.

Phase 1: Understand PyTorch Semantics

Installs
1
GitHub Stars
100
First Seen
5 days ago
gpu-kernel-baseline — alibaba/atrex-kernel-agent