[Spec] trtllm mha supports overlap plan stream (#25925)
This commit is contained in:
@@ -537,6 +537,11 @@ class TRTLLMHAAttnBackend(FlashInferAttnBackend):
|
||||
self._copy_swa_page_table(metadata, page_indices, max_seq_pages)
|
||||
self.forward_metadata = metadata
|
||||
|
||||
def update_verify_buffers_to_fill_after_draft(
|
||||
self, spec_info: SpecInput, cuda_graph_bs: Optional[int]
|
||||
):
|
||||
pass
|
||||
|
||||
def get_cuda_graph_seq_len_fill_value(self) -> int:
|
||||
"""Get the fill value for sequence lengths in CUDA graph."""
|
||||
return 1
|
||||
|
||||
Reference in New Issue
Block a user