From fd16e05252f29cfd9caad34620319bbeb32e745f Mon Sep 17 00:00:00 2001
From: zhaozx-cn <59479021+zhaozx-cn@users.noreply.github.com>
Date: Mon, 1 Jun 2026 21:44:24 +0800
Subject: [PATCH] [NPU] fix npu profiler (#24835)

---
 .../scheduler_components/profiler_manager.py      | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/python/sglang/srt/managers/scheduler_components/profiler_manager.py b/python/sglang/srt/managers/scheduler_components/profiler_manager.py
index 8b446fe28..6e5a58ffa 100644
--- a/python/sglang/srt/managers/scheduler_components/profiler_manager.py
+++ b/python/sglang/srt/managers/scheduler_components/profiler_manager.py
@@ -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