スキル一覧に戻る

analyze-kernel-bottleneck

pjt222
更新日 2 days ago
2 閲覧
17
2
17
GitHubで表示
その他general

について

このスキルは、GPUカーネルを分析し、ルーフライン分析と占有率計算を用いて、演算律速、メモリ律速、レイテンシ律速のいずれかに分類します。最適化戦略(タイリングやダブルバッファリングなど)を推奨するための決定マトリックスを提供します。高度なCUDAカーネルのパフォーマンスプロファイリングとボトルネック特定にご活用ください。

クイックインストール

Claude Code

推奨
メイン
npx skills add pjt222/agent-almanac -a claude-code
プラグインコマンド代替
/plugin add https://github.com/pjt222/agent-almanac
Git クローン代替
git clone https://github.com/pjt222/agent-almanac.git ~/.claude/skills/analyze-kernel-bottleneck

このコマンドをClaude Codeにコピー&ペーストしてスキルをインストールします

ドキュメント

Kernel-Engpass analysieren

Systematisch identifizieren ob ein GPU-Kernel rechen-, speicher- oder latenzbegrenzt ist, indem die Baseline-Performance gemessen, auf der Roofline klassifiziert, Occupancy und Compute-Load-Ratio pro Tile berechnet, SASS-Instruktionsmix und Stall-Codes inspiziert, der Shared-Memory-Cliff geprueft und eine Entscheidungsmatrix angewendet wird um die richtige Optimierungsstrategie zu waehlen.

Wann verwenden

  • Vor der Optimierung jedes CUDA-Kernels -- Baseline etablieren und Engpasstyp klassifizieren
  • Nachdem eine erste funktionierende Version eines Kernels geschrieben wurde, um den Optimierungspfad zu identifizieren
  • Wenn ein Kernel die Erwartungen relativ zum theoretischen Peak unterschreitet
  • Bei der Entscheidung zwischen cp.async, groesseren Tiles oder algorithmischer Restrukturierung

Eingaben

  • Erforderlich: Kompilierter Kernel (.cubin oder .cu-Quelle mit Build-Befehl)
  • Erforderlich: Benchmark-Harness der den Kernel mit CUDA-Event-Timing startet
  • Erforderlich: Problemdimensionen (z.B. M, N, K fuer GEMM; seq_len, heads, head_dim fuer Attention)
  • Optional: Ziel-GPU-Architektur (Standard: GA104 / sm_86 / RTX 3070 Ti)
  • Optional: Erwarteter Peak-Auslastungsprozentsatz fuer Vergleich
  • Optional: Frueher Profiling-Daten (Nsight-Compute-Berichte)

Vorgehensweise

Schritt 1: Baseline-Performance messen

Den Kernel mit CUDA-Events (BenchTimer) ausfuehren, Zeit in Millisekunden aufzeichnen. Effektive Durchsatzmetriken berechnen:

  1. Kompilieren des Kernels falls noch nicht gebaut:
    nvcc --cubin -arch=sm_86 -O2 -o kernel.sm_86.cubin kernel.cu
    nvcc -arch=sm_86 -O2 -o bench bench.cu -lcuda -I../../phase2/common
    
  2. Ausfuehren mit repraesentativen Problemgroessen, sicherstellen dass Warmup-Laeufe der Messung vorausgehen:
    ./bench 4096 4096 4096
    
  3. Aufzeichnen der Kernel-Zeit in ms aus CUDA-Events (nicht Wall-Clock).
  4. Berechnen effektiver GFLOPS und effektiver Bandbreite:
    • GEMM: effective_gflops = (2 * M * N * K) / (time_ms / 1000) / 1e9
    • Bandbreitenbegrenzte Kernels: effective_bw = total_bytes / (time_ms / 1000) / 1e9
    • Flash Attention: effective_gflops = (4 * batch * heads * seq_len^2 * head_dim) / (time_ms / 1000) / 1e9

Erwartet: Baseline-Zahlen: Kernel-Zeit in ms, effektive GFLOPS und effektive Bandbreite.

Bei Fehler: Pruefen dass der Kernel ohne Fehler startet (CHECK_CU-Macro). Verifizieren dass Warmup-Laeufe der Messung vorausgehen. Sicherstellen dass Problemdimensionen gross genug sind um die GPU zu saettigen (kleine Probleme koennen am Launch-Overhead haengen bleiben).

Schritt 2: Auf der Roofline klassifizieren

Arithmetische Intensitaet berechnen und mit dem Maschinen-Balancepunkt vergleichen um den Kernel zu klassifizieren:

  1. Arithmetische Intensitaet berechnen: AI = FLOPs / bytes_loaded_from_global_memory. Nur einzigartige Bytes aus DRAM zaehlen (nicht Shared Memory oder Register-Reuse).
  2. Maschinen-Balancepunkt nachschlagen: balance = peak_compute / peak_bandwidth.
  3. Klassifizieren: Wenn AI < balance, ist der Kernel speicherbegrenzt. Wenn AI > balance, ist der Kernel rechenbegrenzt.

GA104 (RTX 3070 Ti) Referenzwerte:

ResourcePeakUnit
FP32 FFMA21.7TFLOPS
FP16 Tensor Core (HMMA)174TFLOPS
INT8 Tensor Core (IMMA)696TOPS
DRAM Bandwidth608GB/s
L2 Cache4MB
SMs48

Abgeleitete Balancepunkte:

PrecisionBalance Point (FLOP/byte)
FP32 FFMA21700 / 608 = 35.7
FP16 TC174000 / 608 = 286.2
INT8 TC696000 / 608 = 1144.7
  1. Erreichten Anteil berechnen: attained = effective_throughput / peak_throughput. Bei Speicherbegrenzung: effektive Bandbreite mit 608 GB/s vergleichen. Bei Rechenbegrenzung: effektive GFLOPS mit dem relevanten Peak vergleichen.

Erwartet: Klassifikation als rechen-, speicher- oder latenzbegrenzt (geringe Occupancy verursacht weder Rechen- noch Speichersaettigung) mit numerischer Begruendung.

Bei Fehler: Byte-Zaehlung erneut pruefen. Auf redundante Reads achten (z.B. 9x bei direkter conv2d ohne im2col). Wenn weder Rechnung noch Speicher gesaettigt sind, ist der Kernel wahrscheinlich latenzbegrenzt (siehe Schritt 3).

Schritt 3: Occupancy berechnen

Aktive Warps pro SM aus der Launch-Konfiguration und dem Ressourcenverbrauch ermitteln:

  1. Ressourcenverbrauch extrahieren:
    nvcc --cubin -arch=sm_86 -O2 --resource-usage -o kernel.sm_86.cubin kernel.cu 2>&1 | grep -E 'registers|smem'
    
  2. Aus Launch-Konfig: warps_per_block = threads_per_block / 32.
  3. Bloecke/SM berechnen aus jedem begrenzenden Faktor:
    • Register-Limit: floor(65536 / (registers_per_thread * threads_per_block))
    • Smem-Limit: floor(available_smem_per_SM / smem_per_block) -- siehe Schritt 6 fuer Cliff
    • Warp-Limit: floor(48 / warps_per_block) (GA104 max: 48 Warps/SM)
    • Block-Limit: 16 Bloecke/SM max auf GA104
  4. Tatsaechliche Bloecke/SM = min(register_limit, smem_limit, warp_limit, block_limit).
  5. Aktive Warps/SM = blocks_per_SM * warps_per_block.
  6. Schluessel-Schwellwert: 8 Warps/SM sind ausreichend fuer Latency Hiding auf GA104. Unterhalb 8 = strukturelles Problem das latenzbegrenztes Verhalten verursacht.

Erwartet: Occupancy-Tabelle die Bloecke/SM, aktive Warps/SM und den begrenzenden Faktor (Register, smem oder Warps) zeigt.

Bei Fehler: cuFuncSetAttribute fuer dynamisches Shared Memory pruefen. Verifizieren dass --resource-usage-Berichte mit der tatsaechlichen Launch-Konfiguration uebereinstimmen. Wenn Register-Anzahl unerwartet hoch ist, --maxrregcount=N versuchen um Register zu deckeln (Register-Spills gegen Occupancy tauschen).

Schritt 4: Compute-Load-Ratio pro Tile berechnen

Compute-Instruktionen und Load-Bytes pro K-Tile aus SASS zaehlen (nicht Quellcode):

  1. Disassemblieren:
    cuobjdump -sass kernel.sm_86.cubin > kernel.sass
    
  2. Compute-Instruktionen pro Tile zaehlen (die innere Schleife ueber ein K-Tile):
    • grep -c 'HMMA' kernel.sass -- FP16-Tensor-Core-Ops
    • grep -c 'IMMA' kernel.sass -- INT8-Tensor-Core-Ops
    • grep -c 'FFMA' kernel.sass -- FP32-Fused-Multiply-Add
  3. Globale Loads pro Tile zaehlen:
    • grep -c 'LDG' kernel.sass -- Global-Memory-Loads
    • Mit Bytes pro Load multiplizieren (typischerweise 16 Bytes fuer LDG.128)
  4. Ratio berechnen: compute_ops / load_ops pro Tile.
  5. Klassifizieren mit dem cp.async-Entscheidungsschwellwert (aus gpu_reflections.md Insight 2):
    • Hoch (>20:1): cp.async ist netto-negativ; Warp-Interleaving versteckt bereits DRAM-Latenz. Auf algorithmische Aenderungen fokussieren. Referenz: Flash Attention hat 64 HMMA pro Tile = hohes Ratio, cp.async gemessen -5%.
    • Mittel (5-20:1): cp.async kann helfen, beide Pfade benchmarken.
    • Niedrig (<5:1): cp.async stark vorteilhaft; Loads dominieren und Async-Copy versteckt Latenz. Referenz: IGEMM hat 8 IMMA pro Tile = niedriges Ratio, cp.async gemessen +35%.

Erwartet: Compute-Load-Ratio mit Klassifikation (hoch/mittel/niedrig) und cp.async-Empfehlung.

Bei Fehler: Aus SASS-Disassembly zaehlen, nicht Quellcode -- der Compiler kann fusionieren, eliminieren oder Instruktionen umordnen. Sicherstellen dass nur Instruktionen innerhalb der inneren Schleife gezaehlt werden (die K-Tile-Iteration), nicht der ganze Kernel.

Schritt 5: SASS-Instruktionen inspizieren

Den vollstaendigen SASS-Instruktionsmix und Stall-Codes pruefen:

  1. Disassemblieren (falls in Schritt 4 nicht erfolgt):
    cuobjdump -sass kernel.sm_86.cubin > kernel.sass
    
  2. Schluesselinstruktionstypen zaehlen:
    grep -c 'HMMA.16816' kernel.sass      # FP16 Tensor Core
    grep -c 'IMMA.16816' kernel.sass      # INT8 Tensor Core
    grep -c 'FFMA' kernel.sass            # FP32 fused multiply-add
    grep -c 'LDGSTS' kernel.sass          # cp.async (global->shared)
    grep -c 'LDG' kernel.sass             # Global load
    grep -c 'STS' kernel.sass             # Shared store
    grep -c 'LDS' kernel.sass             # Shared load
    grep -c 'BAR.SYNC' kernel.sass        # Barrier synchronization
    grep -c 'SHFL' kernel.sass            # Warp shuffle (reductions)
    grep -c 'MUFU' kernel.sass            # Special function unit
    
  3. Stall-Codes pruefen auf kritischen Instruktionen:
    grep 'HMMA' kernel.sass | head -5     # Expect S08 minimum (hardware constraint)
    grep 'IMMA' kernel.sass | head -5     # Compiler emits S04, reducible to S02 via CuAssembler
    grep 'FFMA' kernel.sass | head -5     # Check for S04 (reducible to S01 on independent FFMAs)
    
  4. Optimierungsziele identifizieren:
    • HMMA-S08-Stalls: Hardware-Minimum auf Ampere, kann nicht reduziert werden. Anderswo fokussieren.
    • IMMA-S04-Stalls: Compiler ist konservativ. CuAssembler kann auf S02 verengen (gemessen 15-20% Gewinn).
    • FFMA-S04-Stalls: bei Unabhaengigkeit auf S01 reduzierbar via CuAssembler.
    • Exzessives BAR.SYNC: kann auf Ueber-Synchronisation zwischen Pipeline-Stufen hinweisen.

Erwartet: Instruktionsanzahltabelle und Stall-Code-Zusammenfassung mit identifizierten Optimierungszielen.

Bei Fehler: Sicherstellen dass die cuobjdump-Architektur mit dem Kernel-Kompilierungsziel uebereinstimmt (beide muessen sm_86 sein). Wenn SASS-Ausgabe leer ist, koennte das cubin korrupt sein -- neu kompilieren.

Schritt 6: Den Smem-Cliff pruefen

Ermitteln ob der Shared-Memory-Verbrauch den architekturspezifischen Occupancy-Cliff ueberschreitet:

  1. Smem/Block lesen aus --resource-usage-Ausgabe (Schritt 3) oder cuobjdump --res-usage kernel.sm_86.cubin.
  2. Mit Cliff-Schwellwert vergleichen:
    • GA104 (sm_86): 100 KB max smem/SM. Cliff bei 50 KB/Block.
    • Empirisch bestaetigt: 48 KB/Block -> 2 Bloecke/SM (gut), 56 KB/Block -> 1 Block/SM (2x Regression).
  3. Wenn ueber Cliff (smem > 50 KB/Block):
    • Bloecke/SM faellt auf 1, aktive Warps fallen auf warps_per_block (typischerweise 4).
    • 2x Performance-Regression erwartet durch exponierte DRAM-Stalls.
  4. Auswirkung von Double-Buffering pruefen: Double-Buffering verdoppelt smem-Verbrauch. Bei aktuellen 30 KB smem ist double-buffered = 60 KB, was den Cliff ueberschreitet. Bewerten ob der Async-Vorteil den Occupancy-Verlust ueberwiegt.
  5. Aufzeichnen smem/Block, Bloecke/SM und ob der Cliff ueberschritten wird.

Erwartet: Smem/Block-Wert mit Bloecke/SM-Anzahl und expliziter Aussage ob der 50-KB-Cliff ueberschritten wird.

Bei Fehler: Wenn ueber Cliff und Occupancy der Engpass ist, muss sich die Optimierungsstrategie aendern: Tile-Groesse reduzieren um smem unter 50 KB zu bekommen, oder 1 Block/SM akzeptieren und mit hoeherer Compute-Load-Ratio pro Tile kompensieren (mehr Register-Reuse, laengere K-Tiles).

Schritt 7: Die Entscheidungsmatrix bauen

Befunde aus Schritten 2-6 zu einer Optimierungsstrategie synthetisieren:

BedingungStrategie
Speicherbegrenzt + niedrige Compute-Load-Ratio (<5:1) + smem unter CliffSoftware-Pipelining mit cp.async (LDGSTS). Globale Loads mit Compute ueberlappen.
Speicherbegrenzt + hohe Compute-Load-Ratio (>20:1) + 8+ WarpsWarp-Interleaving versteckt bereits Latenz. Auf algorithmische Aenderungen fokussieren: implicit GEMM, split-Q, im2col.
Rechenbegrenzt + FFMA-lastigCuAssembler-Stall-Code-Verengung: S04 -> S01 auf unabhaengigen FFMAs.
Rechenbegrenzt + HMMA-lastigS08 ist Hardware-Minimum, nicht reduzierbar. Tile-Reuse erhoehen (groessere M/N-Tiles, laengere K-Schleife).
Rechenbegrenzt + IMMA-lastigCuAssembler: S04 -> S02 auf IMMA-Instruktionen (Compiler ist konservativ).
Latenzbegrenzt (niedrige Occupancy, weder gesaettigt)Smem oder Register reduzieren um mehr Bloecke/SM zu erhalten. Ueber 8 Warps/SM kommen.
Smem ueber CliffTile-Groesse reduzieren oder restrukturieren um smem/Block unter 50 KB zu bekommen (GA104).
  1. Anwendbare Strategien rangieren nach erwartetem Gewinn, mit Compute-Load-Ratio und Occupancy-Daten.
  2. Gewinn-Bereich schaetzen fuer jede Strategie basierend darauf wie weit der Kernel von der relevanten Decke entfernt ist.
  3. Konflikte markieren: z.B. cp.async verdoppelt smem (kann Cliff ueberschreiten), groessere Tiles erhoehen Register-Druck (koennen Occupancy reduzieren).

Erwartet: Rangierte Liste empfohlener Optimierungen mit vorhergesagtem Gewinn-Bereich und potenziellen Konflikten.

Bei Fehler: Wenn kein klarer Gewinner auftaucht, Mikro-Benchmarks ausfuehren die jede Strategie isolieren (z.B. cp.async allein testen, reduzierte Tile-Groesse allein testen) um die tatsaechliche Wirkung vor dem Kombinieren zu messen.

Schritt 8: Befunde dokumentieren

Einen strukturierten Engpass-Bericht erzeugen:

  1. Baseline: Kernel-Zeit, effektive GFLOPS, effektive Bandbreite, Problemdimensionen.
  2. Roofline-Position: arithmetische Intensitaet, Klassifikation, erreichter Peak-Anteil.
  3. Occupancy: Bloecke/SM, aktive Warps/SM, begrenzender Faktor.
  4. Compute-Load-Ratio: Ratio-Wert, Klassifikation (hoch/mittel/niedrig), cp.async-Empfehlung.
  5. SASS-Zusammenfassung: Instruktionsanzahltabelle, Stall-Code-Befunde, CuAssembler-Ziele.
  6. Smem-Cliff: smem/Block, Bloecke/SM, Cliff-Status.
  7. Empfehlung: rangierte Optimierungsstrategien mit Gewinn-Schaetzungen.
## Bottleneck Analysis Report: [kernel_name]

### Baseline
- Problem: [dimensions]
- Kernel time: [X] ms
- Effective GFLOPS: [Y] | Effective BW: [Z] GB/s

### Roofline Classification
- Arithmetic intensity: [AI] FLOP/byte
- Balance point: [BP] FLOP/byte ([precision])
- Classification: **[compute|memory|latency]-bound**
- Attained fraction: [X]% of peak

### Occupancy
| Resource | Per Block | Limit/SM | Blocks/SM |
|----------|-----------|----------|-----------|
| Registers | [N]/thread | 65536 | [B] |
| Shared mem | [X] KB | 100 KB (cliff: 50 KB) | [B] |
| Warps | [W] | 48 | [B] |
| **Limiting** | | | **[min(B)]** |
- Active warps/SM: [W] ([sufficient|insufficient] for latency hiding)

### Compute/Load Ratio
- Compute ops/tile: [N] [HMMA|IMMA|FFMA]
- Load bytes/tile: [N] bytes ([N] LDG x [N] bytes)
- Ratio: [X]:1 — **[high|medium|low]**
- cp.async recommendation: [beneficial|neutral|detrimental]

### SASS Instruction Mix
| Instruction | Count | Notes |
|-------------|-------|-------|
| HMMA.16816 | [N] | Stall: S08 (hardware min) |
| IMMA.16816 | [N] | Stall: S04 (reducible to S02) |
| FFMA | [N] | Stall: S04 (reducible to S01) |
| LDG | [N] | |
| LDGSTS | [N] | cp.async |
| BAR.SYNC | [N] | |

### Smem Cliff
- Smem/block: [X] KB — [under|over] 50 KB cliff
- Blocks/SM: [B] — [no occupancy loss|occupancy halved]

### Recommended Optimizations (ranked)
1. [Strategy] — estimated [X-Y]% gain
2. [Strategy] — estimated [X-Y]% gain
3. [Strategy] — estimated [X-Y]% gain

Erwartet: Vollstaendiger Markdown-Bericht der von einem Kernel-Optimizer-Agenten oder menschlichen Entwickler konsumierbar ist.

Bei Fehler: Mit unterschiedlichen Problemgroessen erneut ausfuehren (z.B. 1024, 2048, 4096, 8192) um zu bestaetigen dass Befunde nicht groessenspezifisch sind. Kleine Probleme koennen latenzbegrenzt erscheinen waehrend der echte Engpass im Massstab Speicherbandbreite ist.

Validierung

  • Baseline mit CUDA-Events gemessen (nicht Wall-Clock)
  • Roofline-Klassifikation ermittelt (rechen-/speicher-/latenzbegrenzt)
  • Occupancy berechnet mit identifiziertem begrenzendem Faktor
  • Compute-Load-Ratio pro Tile aus SASS berechnet
  • SASS-Instruktionsmix und Stall-Codes dokumentiert
  • Smem-Cliff gegen Architektur-Schwellwert geprueft
  • Entscheidungsmatrix angewendet mit Strategie-Empfehlung
  • Befunde in strukturiertem Bericht dokumentiert

Haeufige Stolperfallen

  • Re-Read-Multiplikation: Direkte conv2d liest jedes Gewicht 9x ohne im2col, blaeht die Byte-Anzahl um 9x auf. Tatsaechliche einzigartige aus DRAM geladene Bytes nutzen, nicht gesamte Load-Instruktionen, beim Berechnen arithmetischer Intensitaet.
  • FP16-Tensor-Core-Peak mit FP32-Peak verwechseln: FP16-TC-Peak ist 174 TFLOPS, FP32-FFMA-Peak ist 21,7 TFLOPS -- ein 8x-Unterschied. Mit dem falschen Peak wird Roofline-Klassifikation bedeutungslos.
  • 64 KB als smem-Cliff statt 50 KB auf GA104 nutzen: GA104 (sm_86) hat 100 KB max smem/SM. Der Cliff ist bei 100/2 = 50 KB/Block, nicht 64 KB. Dies ist architekturspezifisch; andere GPUs unterscheiden sich.
  • Warp-Interleaving beim Bewerten von cp.async ignorieren: 8 Warps mit langen Compute-Phasen (hohe Compute-Load-Ratio) verstecken bereits DRAM-Latenz durch Warp-Scheduling. cp.async in diesem Regime hinzuzufuegen fuegt smem-Druck und Barrier-Overhead ohne Nutzen hinzu (gemessen -5% auf Flash Attention).
  • Instruktionen aus Quellcode statt SASS zaehlen: Der Compiler kann Operationen fusionieren, toten Code eliminieren, Schleifen anders unrollen oder Instruktionen umordnen. Immer aus cuobjdump -sass-Ausgabe zaehlen.
  • Keine Warmup-Iterationen ausfuehren: Der erste Kernel-Launch enthaelt JIT-Compile-Overhead und Cold-Cache-Effekte. Immer 2-5 Warmup-Iterationen vor dem gemessenen Lauf ausfuehren.

Verwandte Skills

  • pipeline-gpu-kernel -- Software-Pipelining mit cp.async implementieren wenn die Analyse einen speicherbegrenzten Kernel mit niedriger Compute-Load-Ratio identifiziert
  • simulate-cpu-architecture -- ergaenzende Architekturanalyse fuer CPU-seitige Engpaesse in Host-Device-Workflows

GitHub リポジトリ

pjt222/agent-almanac
パス: i18n/de/skills/analyze-kernel-bottleneck
0
agentsagentskillsai-assisted-developmentclaude-codeskillsteams

関連スキル

llamaguard

その他

LlamaGuardは、暴力やヘイトスピーチなど6つの安全性カテゴリーにおいて、LLMの入力と出力をモデレートするMetaの70-80億パラメータモデルです。94〜95%の精度を提供し、vLLM、Hugging Face、Amazon SageMakerを使用してデプロイ可能です。このスキルを使用して、AIアプリケーションにコンテンツフィルタリングと安全策を簡単に統合できます。

スキルを見る

cost-optimization

その他

このClaudeスキルは、リソースの適正サイジング、タグ付け戦略、支出分析を通じて、開発者がクラウドコストを最適化することを支援します。AWS、Azure、GCPにわたるクラウド支出の削減とコストガバナンスの実施のためのフレームワークを提供します。インフラコストの分析、リソースの適正サイジング、または予算制約への対応が必要な際にご利用ください。

スキルを見る

quantizing-models-bitsandbytes

その他

このスキルは、bitsandbytesを使用してLLMを8ビットまたは4ビット精度に量子化し、精度の低下を最小限に抑えつつ50〜75%のメモリ削減を実現します。限られたGPUメモリでより大規模なモデルを実行したり、推論を高速化するのに理想的で、INT8、NF4、FP4などのフォーマットをサポートしています。HuggingFace Transformersと統合され、QLoRAトレーニングや8ビットオプティマイザーを可能にします。

スキルを見る

dispatching-parallel-agents

その他

このClaudeスキルは、複数のエージェントを配備し、3つ以上の独立した問題を並行して調査・修正します。共有状態や依存関係がなく解決可能な、無関係な障害が発生するシナリオ向けに設計されています。中核となる機能は並列問題解決であり、効率を最大化するために独立した問題領域ごとに1つのエージェントを割り当てます。

スキルを見る