跳到正文
Hugging Face Blog·· 12 天前AI 评分28

TorchNeuron 性能剖析(一):在 AWS Trainium 上做系统级性能分析

Profiling TorchNeuron (Part 1): System-level profiling on AWS Trainium.

AI 导读

系列文章第一部分讲解如何用 torch.profiler 配合 NeuronConfig 与 NeuronProfiler 在 AWS Trainium 上系统分析推理性能,捕获干净的 trace 并导出 trace.json。

来源:Hugging Face Blog · huggingface.co