[NPU] [bugfix] Fix NPU MLA HiCache backup accessing missing data_ptrs. (#36813)

2026-09-01 17:31:11 +08:00
parent dc1ae02684
commit c16a8fc899
1 changed files with 8 additions and 3 deletions
@@ -441,9 +441,14 @@ class MLATokenToKVPoolHost(HiSparseHostPoolMixin, HostKVCache):
                )
            return

-        device_data_ptrs, device_kv_buffers = self._resolve_device_transfer_buffers(
-            device_pool
-        )
+        if io_backend == "kernel_ascend":
+            # NPU pools use contiguous multi-layer tensors and intentionally do
+            # not build the CUDA-style data_ptrs array.
+            device_data_ptrs, device_kv_buffers = None, None
+        else:
+            device_data_ptrs, device_kv_buffers = self._resolve_device_transfer_buffers(
+                device_pool
+            )

        if io_backend == "kernel":
            if self.layout == "layer_first":