torch_npu.profiler.ProfilerActivity
API接口
torch_npu.profiler.ProfilerActivity
功能描述
事件采集列表,枚举类。用于赋值给torch_npu.profiler.profile的activities参数。
参数说明
torch_npu.profiler.ProfilerActivity.CPU:框架侧数据采集的开关。
torch_npu.profiler.ProfilerActivity.NPU:CANN软件栈及NPU数据采集的开关。
父主题:
torch_npu.profiler