gpu-kernel-profile-optimizer

Profile GPU kernels with NCU or ROCprof and iteratively optimize bottlenecks.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

This skill solves the challenge of performance-tuning GPU kernels by replacing intuition-based guessing with a rigorous, evidence-based optimization loop that identifies and resolves hardware-level bottlenecks.

Core Features & Use Cases

  • Evidence-Driven Optimization: Uses profiling tools like NCU and ROCprof to pinpoint exact performance bottlenecks such as memory throughput, occupancy, or bank conflicts.
  • Structured Iterative Workflow: Manages the entire lifecycle of kernel optimization, including profiling, bottleneck analysis, plan generation, implementation, and validation.
  • Use Case: When a PyTorch operator is underperforming, use this skill to profile the kernel, identify a specific warp stall reason, and apply a targeted optimization like memory coalescing or register pressure reduction.

Quick Start

Use the gpu-kernel-profile-optimizer skill to begin an iterative optimization session for the kernel file located in the current workspace.

Frequently Asked Questions about gpu-kernel-profile-optimizer

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

FAQPage Schema
How do I identify GPU kernel performance bottlenecks using profiling?

GPU kernel performance bottlenecks are identified using profiling tools like NCU and ROCprof to pinpoint exact issues such as memory throughput limits, low occupancy, or bank conflicts. This approach replaces intuition-based guessing with rigorous, evidence-based bottleneck analysis.

What is the best way to optimize a slow PyTorch operator on GPU hardware?

The best way to optimize a slow PyTorch operator is through a structured, profile-driven iterative workflow. This involves profiling the kernel, identifying specific warp stall reasons, and applying targeted optimizations like memory coalescing or register pressure reduction.

Does this profile-driven optimization workflow support both NVIDIA and AMD hardware?

Yes, this profile-driven optimization workflow supports high-performance computing tasks on both NVIDIA and AMD hardware. It utilizes specific profiling tools like NCU for NVIDIA and ROCprof for AMD to measure performance and validate systematic code refinement.

How do I start an iterative GPU kernel optimization session for a specific file?

To start an iterative GPU kernel optimization session, you provide the kernel file located in your current workspace to the skill. It then manages the entire lifecycle, including profiling, bottleneck analysis, plan generation, implementation, and validation.

Why should I use an evidence-based optimization loop instead of intuition for GPU kernels?

An evidence-based optimization loop is necessary because it resolves hardware-level bottlenecks through rigorous validation and performance measurement. This structured cycle generates auditable evidence, ensuring systematic code refinement rather than relying on intuition-based guessing.

What types of hardware-level bottlenecks can GPU kernel profiling pinpoint?

GPU kernel profiling can pinpoint exact hardware-level bottlenecks such as memory throughput limitations, insufficient occupancy, and bank conflicts. Identifying these specific issues allows for targeted code refinement during the optimization process.