[NPU] Fix NPU CI (#13834)

Co-authored-by: c30031083 <chenxu140@huawei.com>
2025-11-25 10:09:36 +08:00
parent eb1d885400
commit db0ffc09ef
8 changed files with 11 additions and 31 deletions
--- a/python/sglang/srt/speculative/spec_utils.py
+++ b/python/sglang/srt/speculative/spec_utils.py
@@ -133,8 +133,8 @@ def assign_req_to_token_pool_func(
        torch.ops.npu.cache_loc_assign(
            req_pool_indices,
            req_to_token,
-            start_offset,
-            end_offset,
+            start_offset.to(torch.int64),
+            end_offset.to(torch.int64),
            out_cache_loc,
        )