跳到主要内容

CUDA

带「CUDA」标签的杂记 · 共 2 篇

Triton 入门:写第一个 kernel

用 Triton 重写向量加法,理解 program、tl.arange 与 block 启动背后的模型。

GPU 性能分析工具地图

nvidia-smi、PyTorch Profiler、Nsight Compute 各自看什么,先看哪个指标。