Unverified Commit fc809665 authored by Jinyan Chen's avatar Jinyan Chen Committed by GitHub
Browse files

[Performance] qwen3-next improve causal conv1d in prefill phase (#10595)


Co-authored-by: default avatarJinyan Chen <jinyanc@nvidia.com>
parent 6fd4816d
......@@ -13,6 +13,9 @@ from sglang.srt.layers.attention.fla.fused_recurrent import (
from sglang.srt.layers.attention.fla.fused_sigmoid_gating_recurrent import (
fused_sigmoid_gating_delta_rule_update,
)
from sglang.srt.layers.attention.mamba.causal_conv1d import (
causal_conv1d_fn as causal_conv1d_fn_sgl,
)
from sglang.srt.layers.attention.mamba.causal_conv1d_triton import (
causal_conv1d_fn,
causal_conv1d_update,
......@@ -334,7 +337,7 @@ class MambaAttnBackend(AttentionBackend):
mixed_qkv_processed.transpose(1, 2).contiguous().view(seq_len, -1)
)
else:
mixed_qkv = causal_conv1d_fn(
mixed_qkv = causal_conv1d_fn_sgl(
mixed_qkv.transpose(0, 1),
conv_weights,
bias,
......
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