AgentSkillsCN

ir-debugging

通过在每个阶段（TTIR、TTGIR、LLVM、PTX）转储IR，调试Triton编译过程。当您需要调查编译失败、内核性能、寄存器溢出，或用户希望查看IR输出时，可调用此功能。涵盖TRITON_KERNEL_DUMP、MLIR_ENABLE_DUMP、LLVM_IR_ENABLE_DUMP、TRITON_DUMP_PTXAS_LOG，以及相关环境变量。

SKILL.md
--- frontmatter
name: ir-debugging
description: >
  Debug Triton compilation by dumping IR at each stage (TTIR, TTGIR, LLVM, PTX).
  Use when investigating compilation failures, kernel performance, register
  spills, or when user asks to inspect IR output. Covers TRITON_KERNEL_DUMP,
  MLIR_ENABLE_DUMP, LLVM_IR_ENABLE_DUMP, TRITON_DUMP_PTXAS_LOG, and related env vars.

IR Debugging

Environment variables

Env varWhat it does
TRITON_KERNEL_DUMP=1Dump IR at every compilation stage to ~/.triton/dump/
TRITON_PRINT_AUTOTUNING=1Use human-readable per-config subdirectories instead of hashes (combine with KERNEL_DUMP)
TRITON_KERNEL_DUMP_BEST_CONFIG=1Dump IR only for the winning autotuned config (re-compiles with dumping, avoids noise)
MLIR_ENABLE_DUMP=1Dump MLIR IR during pass execution (filter by kernel: MLIR_ENABLE_DUMP=_kernel)
LLVM_IR_ENABLE_DUMP=1Dump LLVM IR (print-after-all)
NVPTX_ENABLE_DUMP=1Dump NVPTX backend IR
TRITON_DUMP_PTXAS_LOG=1Dump ptxas assembler logs (register usage, spills)
TRITON_INTERPRET=1Run kernels in interpreter mode (no GPU needed)
TRITON_ALWAYS_COMPILE=1Bypass cache, force recompilation
TRITON_DUMP_TTGIR_TO_TLX=1Dump TTGIR back to TLX Python (reverse-engineer IR)

Decision tree: what are you debugging?

  • •"Kernel produces wrong results" → TRITON_INTERPRET=1 to run on CPU, or TRITON_KERNEL_DUMP=1 to inspect IR at each stage
  • •"Kernel is slow / register spills" → TRITON_DUMP_PTXAS_LOG=1 to check register usage and spills
  • •"Which autotuned config won and why?" → TRITON_KERNEL_DUMP_BEST_CONFIG=1 TRITON_PRINT_AUTOTUNING=1
  • •"Need to see MLIR passes" → MLIR_ENABLE_DUMP=1 (optionally filter: MLIR_ENABLE_DUMP=_my_kernel)
  • •"Need to see final PTX/LLVM" → LLVM_IR_ENABLE_DUMP=1 and/or NVPTX_ENABLE_DUMP=1
  • •"Cached result is stale" → TRITON_ALWAYS_COMPILE=1 to force recompilation

Common combos

bash
# Full dump of best config with readable directory names
TRITON_KERNEL_DUMP_BEST_CONFIG=1 TRITON_PRINT_AUTOTUNING=1 python my_kernel.py

# Debug register pressure
TRITON_DUMP_PTXAS_LOG=1 TRITON_ALWAYS_COMPILE=1 python my_kernel.py

# Inspect MLIR passes for a specific kernel
MLIR_ENABLE_DUMP=_my_kernel TRITON_ALWAYS_COMPILE=1 python my_kernel.py

# Full IR pipeline dump
TRITON_KERNEL_DUMP=1 TRITON_ALWAYS_COMPILE=1 python my_kernel.py

Reference files

  • •Full Python knobs: python/triton/knobs.py
  • •C++ env vars: include/triton/Tools/Sys/GetEnv.hpp