torch_npu.profiler.ProfilerAction
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
用于控制Profiler行为状态，如性能数据采集、性能数据采集预热、性能数据采集并保存，Enum类型。
函数原型
[object Object]
参数说明
- torch_npu.profiler.ProfilerAction.NONE：可选参数，无任何行为。
- torch_npu.profiler.ProfilerAction.WARMUP：可选参数，性能数据采集预热。
- torch_npu.profiler.ProfilerAction.RECORD：可选参数，性能数据采集。
- torch_npu.profiler.ProfilerAction.RECORD_AND_SAVE：可选参数，性能数据采集并保存。
返回值说明
无
调用示例
以下是关键步骤的代码示例，不可直接拷贝编译运行，仅供参考。
[object Object]