[GDN] remove XPU path of causal_conv1d_fn and causal_conv1d_update (#35775)
This commit is contained in:
@@ -43,11 +43,6 @@ if is_cuda():
|
|||||||
)
|
)
|
||||||
|
|
||||||
causal_conv1d_fn = causal_conv1d_fn_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():
|
elif is_npu():
|
||||||
from sgl_kernel_npu.fla.fused_gdn_gating import fused_gdn_gating_npu
|
from sgl_kernel_npu.fla.fused_gdn_gating import fused_gdn_gating_npu
|
||||||
from sgl_kernel_npu.mamba.causal_conv1d import (
|
from sgl_kernel_npu.mamba.causal_conv1d import (
|
||||||
|
|||||||
Reference in New Issue
Block a user