fla-nvidia-performance

Profile and benchmark NVIDIA GPU kernels in the FLA repository.

5.5k|643|Updated Dec 20, 2023
One-click install
npx skills add https://github.com/fla-org/flash-linear-attention --skill fla-nvidia-performance
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: fla-nvidia-performance
Source: https://github.com/fla-org/flash-linear-attention/tree/main/.agents/skills/fla-nvidia-performance
Command: npx skills add https://github.com/fla-org/flash-linear-attention --skill fla-nvidia-performance

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires ncu, python, and includes scripts (resource) components.

What problem does it solve?

This Skill provides guidelines and tools for optimizing the performance of NVIDIA GPU kernels in the Flash Linear Attention (FLA) repository, focusing on profiling, hardware baselines, and performance evidence requirements.

Core Features & Use Cases

  • Performance Profiling: Offers detailed profiling using Nsight Compute for GPU kernels.
  • Hardware Baselines: Establishes minimum and preferred hardware configurations for performance testing.
  • Benchmarking: Provides scripts for benchmarking training and generation throughput.
  • Use Case: A developer working on the FLA repository needs to optimize a kernel's performance. This Skill guides them through the profiling process, ensuring that their changes are performant and meet the required benchmarks.

Quick Start

Run the benchmark_training_throughput.py script with your specific workload to evaluate performance improvements after kernel optimizations.

Frequently Asked Questions about fla-nvidia-performance

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

FAQPage Schema
How do I profile NVIDIA GPU kernel performance in the Flash Linear Attention repository?

You can profile NVIDIA GPU kernel performance in the Flash Linear Attention (FLA) repository using Nsight Compute, which provides detailed metrics required for analyzing and optimizing kernel throughput.

What's the best way to benchmark training throughput after kernel optimization?

The best way to benchmark training throughput after kernel optimization is to run the benchmark_training_throughput.py script with your specific workload to evaluate performance improvements and collect evidence.

Do I need specific hardware baselines for GPU performance testing?

Yes, GPU performance testing requires establishing minimum and preferred hardware baseline configurations to ensure your kernel optimizations are accurately evaluated against consistent hardware standards.

Can I use Nsight Compute to establish performance evidence for kernel optimizations?

Yes, Nsight Compute is used for detailed profiling to generate the performance evidence required to validate that your kernel optimizations meet the established throughput benchmarks.

Why does my Flash Linear Attention kernel optimization lack performance evidence?

Your kernel optimization lacks performance evidence because it was not profiled using Nsight Compute against established hardware baselines to verify training and generation throughput improvements.

What dependencies are required to benchmark GPU kernel performance in FLA?

To benchmark GPU kernel performance in FLA, you need Nsight Compute (ncu) for profiling and Python to execute the benchmarking scripts that measure training and generation throughput.