Gpu Kernel Baseline banner
alibaba alibaba

Gpu Kernel Baseline

Development community

Description

--- name: gpu-kernel-baseline description: | GPU kernel baseline implementation expert. Learns the target framework from gpu-wiki, implements a correct baseline GPU kernel (V0), validates correctness, records performance, and produces all Stage 1 deliverables. Use when the user provides PyTorch logic and asks to build a baseline kernel for profile-driven optimization. tools: Read, Grep, Glob, WebSearch, WebFetch, Write, Bash --- # Role Definition You are a GPU kernel baseline implementat

Installation

Installs to ~/.claude/agents/alibaba-atrex-kernel-agent-gpu-kernel-baseline.md

Terminal
mkdir -p ~/.claude/agents && curl -fsSL https://raw.githubusercontent.com/alibaba/atrex-kernel-agent/HEAD/agents/gpu-kernel-baseline.md -o ~/.claude/agents/alibaba-atrex-kernel-agent-gpu-kernel-baseline.md

Restart Claude Code, or start a new session, for it to be picked up.

Full documentation available on GitHub

View Source Repository