[NPU][Bugfix] Disable pinned memory to fix DeepSeek-V2 DP-attention hang (#39423)

2026-09-15 14:59:31 +08:00
parent 575163ff32
commit 8a20062652
2 changed files with 7 additions and 6 deletions
@@ -56,9 +56,7 @@ class NPUDeviceMixin(DeviceMixin):
        return torch.npu.mem_get_info(device_id)

    def is_pin_memory_available(self, device=None) -> bool:
-        if device is not None and str(device) == "cpu":
-            return False
-        return True
+        return False

    @classmethod
    def seed_everything(cls, seed: int | None = None) -> None: