From a2935ce329bc139701d4837ee5cd50d58898e538 Mon Sep 17 00:00:00 2001 From: jiayisunx Date: Thu, 23 Jul 2026 10:50:20 +0800 Subject: [PATCH] [XPU][GDN] add XPU path for causal_conv1d_fn and causal_conv1d_update (#31250) Co-authored-by: Ma Mingfei --- python/sglang/srt/layers/attention/linear/gdn_backend.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/attention/linear/gdn_backend.py b/python/sglang/srt/layers/attention/linear/gdn_backend.py index d3dd717be..61943b885 100644 --- a/python/sglang/srt/layers/attention/linear/gdn_backend.py +++ b/python/sglang/srt/layers/attention/linear/gdn_backend.py @@ -19,7 +19,7 @@ from sglang.srt.layers.radix_linear_attention import RadixLinearAttention from sglang.srt.mem_cache.memory_pool import MambaPool from sglang.srt.model_executor.forward_batch_info import ForwardBatch from sglang.srt.model_executor.model_runner import ModelRunner -from sglang.srt.utils import is_cpu, is_cuda, is_hip, is_npu +from sglang.srt.utils import is_cpu, is_cuda, is_hip, is_npu, is_xpu from sglang.srt.utils.common import rank0_log if not is_cpu(): @@ -40,6 +40,11 @@ if is_cuda(): ) causal_conv1d_fn = causal_conv1d_fn_cuda +elif is_xpu(): + from sgl_kernel import causal_conv1d_fn_xpu, causal_conv1d_update_xpu + + causal_conv1d_fn = causal_conv1d_fn_xpu + causal_conv1d_update = causal_conv1d_update_xpu elif is_npu(): from sgl_kernel_npu.fla.fused_gdn_gating import fused_gdn_gating_npu from sgl_kernel_npu.mamba.causal_conv1d import (