Revert "[feature] Ascend NPU graph support (#8027)" (#9348)

2025-08-20 01:11:23 +08:00
parent 01d47a27b6
commit f4fafacc5d
18 changed files with 878 additions and 1349 deletions
--- a/python/sglang/srt/mem_cache/memory_pool.py
+++ b/python/sglang/srt/mem_cache/memory_pool.py
@@ -376,7 +376,7 @@ class MHATokenToKVPool(KVCache):
        v_scale: Optional[float] = None,
        layer_id_override: Optional[int] = None,
    ):
-        from sglang.srt.model_executor.graph_runner import get_is_capture_mode
+        from sglang.srt.model_executor.cuda_graph_runner import get_is_capture_mode

        if layer_id_override is not None:
            layer_id = layer_id_override