Commit 5e77b44f authored by 王敏's avatar 王敏
Browse files

恢复误删代码

parent 586f0eba
...@@ -5538,6 +5538,10 @@ class GPUModelRunner( ...@@ -5538,6 +5538,10 @@ class GPUModelRunner(
ValueError: If no valid block size found ValueError: If no valid block size found
""" """
#exclude indexer backend
def _participates_in_block_size_selection(backend: type[AttentionBackend]) -> bool:
return not getattr(backend, "exclude_from_block_size_selection", False)
def block_size_is_supported( def block_size_is_supported(
backends: list[type[AttentionBackend]], block_size: int backends: list[type[AttentionBackend]], block_size: int
) -> bool: ) -> bool:
......
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