profiler/ 子模块 — 性能分析工具¶
概述¶
profiler/ 子模块提供统一的性能分析框架，支持 NVIDIA Nsight Systems (NVTX)、华为 NPU (MSTX)、PyTorch Profiler 和 CUDA 内存快照等多种分析后端。
架构¶
DistProfiler (调度器)
├── NsightSystemsProfiler (NVIDIA GPU)
├── NPUProfiler (华为 NPU)
├── Profiler (PyTorch 通用)
├── TorchMemoryProfiler (内存快照)
└── _NoOpProfiler (空操作)
文件列表¶
| 文件 | 说明 |
|---|---|
__init__.py |
入口，自动选择标记实现 |
config.py |
各后端配置 dataclass |
profile.py |
DistProfiler 核心调度器 |
nvtx_profile.py |
NVIDIA NVTX 标记 |
mstx_profile.py |
华为 MSTX 标记 |
torch_profile.py |
PyTorch Profiler |
performance.py |
计时、显存日志 |
empty_annotations.py |
空标记 (fallback) |