[NPU] fix profiler on npu (#24685)

Signed-off-by: zhaozx-cn <zhaozx2116@163.com>
2026-05-09 17:48:24 +08:00
parent fd636410a2
commit e2527df8b6
1 changed files with 15 additions and 0 deletions
@@ -201,6 +201,21 @@ class SchedulerProfilerMixin:
                        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.AiCoreNone,
+                        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