[NPU] fix npu profiler (#24835)

2026-06-01 21:44:24 +08:00
parent 6965fe0eec
commit fd16e05252
1 changed files with 15 additions and 0 deletions
@@ -213,6 +213,21 @@ class SchedulerProfilerManager:
                        str(self.torch_profiler_output_dir)
                    )
                ),
                experimental_config=(
                    None
                    if not _is_npu
                    else torch_npu.profiler._ExperimentalConfig(
                        export_type=torch_npu.profiler.ExportType.Text,
                        profiler_level=torch_npu.profiler.ProfilerLevel.Level1,
                        msprof_tx=False,
                        aic_metrics=torch_npu.profiler.AiCMetrics.PipeUtilization,
                        l2_cache=False,
                        op_attr=False,
                        data_simplification=False,
                        record_op_args=False,
                        gc_detect_threshold=None,
                    )
                ),
            )
            self.torch_profiler.start()
            self.profile_in_progress = True