biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

Browse Source 
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
Michael Goin
2025-09-23 18:48:40 -04:00
committed by GitHub
parent 95bc60e4cb
commit 7361ab379f
28 changed files with 9 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/layers/mamba/short_conv.py
View File

@@ -21,7 +21,6 @@ from vllm.model_executor.layers.mamba.mamba_utils import (
MambaStateDtypeCalculator, MambaStateShapeCalculator)
from vllm.model_executor.layers.mamba.ops.causal_conv1d import (
causal_conv1d_fn, causal_conv1d_update)
from vllm.platforms import current_platform
from vllm.utils import direct_register_custom_op
from vllm.v1.attention.backends.short_conv_attn import (
ShortConvAttentionMetadata)
@@ -251,5 +250,4 @@ direct_register_custom_op(
op_func=short_conv,
mutates_args=["output"],
fake_impl=short_conv_fake,
dispatch_key=current_platform.dispatch_key,
)