[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