gpu-kernel-baseline

Implement and validate baseline GPU kernels using CuteDSL or FlyDSL.

71|22|Updated Jun 23, 2026
One-click install
npx skills add https://github.com/alibaba/atrex-kernel-agent --skill gpu-kernel-baseline
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: gpu-kernel-baseline
Source: https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline
Command: npx skills add https://github.com/alibaba/atrex-kernel-agent --skill gpu-kernel-baseline

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes assets (resource) components.

What problem does it solve?

This skill addresses the complexity of implementing high-performance GPU kernels by providing a structured, profile-driven workflow that ensures correctness and establishes a performance baseline before optimization begins.

Core Features & Use Cases

  • Framework-Specific Implementation: Guides the creation of kernels using CuteDSL or FlyDSL based on target hardware.
  • Automated Validation: Orchestrates correctness testing against PyTorch references with configurable timeout guards.
  • Performance Benchmarking: Automates the calculation of TFLOPS and bandwidth utilization to create a verifiable baseline report.

Quick Start

Use the gpu-kernel-baseline skill to analyze the provided PyTorch logic and generate a validated baseline kernel implementation for the target GPU platform.

Frequently Asked Questions about gpu-kernel-baseline

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

FAQPage Schema
How do I build and validate a baseline GPU kernel from PyTorch logic?▼

To build a baseline GPU kernel from PyTorch logic, you analyze the existing code, generate framework-specific implementations using CuteDSL or FlyDSL, and validate correctness through automated testing against PyTorch references.

What is the best way to benchmark GPU kernel performance and establish a TFLOPS baseline?▼

The best way to benchmark GPU kernel performance is to automate the calculation of TFLOPS and bandwidth utilization, producing a verifiable baseline report that ensures hardware-compliant implementation before optimization begins.

Can I use CuteDSL or FlyDSL to transition my PyTorch code to high-performance GPU kernels?▼

Yes, you can use CuteDSL or FlyDSL to transition PyTorch code to high-performance GPU kernels through structured semantic analysis, ensuring the new implementation matches the original logic before establishing a performance baseline.

Do I need a local gpu-wiki knowledge base to implement high-performance GPU kernels?▼

Yes, integrating a local gpu-wiki knowledge base is required to ensure hardware-compliant implementation and to generate accurate performance reporting when building high-performance GPU kernels.

How does automated correctness testing work for newly generated GPU kernels?▼

Automated correctness testing works by orchestrating direct comparisons against PyTorch references, utilizing configurable timeout guards to safely validate the newly generated GPU kernel output without hanging.