Hugging Face Blog·· 3 小时前精选AI 评分61
TorchNeuron 性能分析(第一部分):AWS Trainium 上的系统级分析
Profiling TorchNeuron (Part 1): System-level profiling on AWS Trainium.
AI 导读
TorchNeuron 性能分析教程以 RMSNorm 为例,演示在 AWS Trainium 上配置 torch.profiler、解读 key_averages(),并追踪算子从框架调度到 NeuronCore 硬件执行。
来源:Hugging Face Blog · huggingface.co