"examples/pytorch/vscode:/vscode.git/clone" did not exist on "fae26dd15caac92458a08ad34889086e1e333ddd"
Unverified Commit 5626e20b authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Tiny fix CI (#9306)

parent c6c379ab
......@@ -66,17 +66,6 @@ def alloc_extend_kernel_ascend(
class AscendPagedTokenToKVPoolAllocator(PagedTokenToKVPoolAllocator):
def __init__(
self,
size: int,
page_size: int,
dtype: torch.dtype,
device: str,
kvcache: KVCache,
need_sort: bool,
):
super().__init__(size, page_size, dtype, device, kvcache, need_sort, 1)
def alloc_extend(
self,
prefix_lens: torch.Tensor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment