TorchNeuron 剖析(第 1 部分):在 AWS Trainium 上进行系统级性能剖析
TorchNeuron 系列教程第 1 部分演示在 AWS Trainium 上用 torch.profiler 捕获 trace,并通过 key_averages() 定位占用 Neuron 设备时间的算子。
TorchNeuron 系列教程第 1 部分演示在 AWS Trainium 上用 torch.profiler 捕获 trace,并通过 key_averages() 定位占用 Neuron 设备时间的算子。