Hugging Face Blog·· 4 小时前AI 评分38
TorchNeuron 性能分析(第 1 部分):AWS Trainium 上的系统级性能分析
Profiling TorchNeuron (Part 1): System-level profiling on AWS Trainium.
AI 导读
Hugging Face 介绍了在 AWS Trainium 芯片上使用开源 PyTorch 原生后端 TorchNeuron 进行系统级性能分析的方法。开发者可通过将 torch.profiler 与 NeuronConfig、NeuronProfiler 配合,捕获设备时间并利用 Neuron Explorer 查看硬件执行轨迹。
来源:Hugging Face Blog · huggingface.co