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.