跳到正文
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