安橙的博客
  • 博文
  • 标签
  • 搜索

Performance

  • Day 26 · 算子层加速

    2026-06-20AIPyTorchCUDA

    进入 Transformer 算子层加速:理解 FlashAttention v2/v3、PyTorch SDPA、xFormers 与 Apex Fused Kernels 的适用边界,动手替换 attention,并用 TFLOPS / MFU 判断优化是否真的生效。

  • Day 27 · 训练性能分析

    2026-06-20AIPyTorchCUDA

    学习训练性能分析的最小闭环:计算 MFU / HFU,用 Nsight Systems 抓一段训练 step,通过 NVTX、CUDA kernel、NCCL timeline 识别 compute、communication 与 pipeline bubble。

© 2026 安橙的博客