diff --git a/python/sglang/srt/custom_op.py b/python/sglang/srt/custom_op.py index c35790691..d770e9c08 100644 --- a/python/sglang/srt/custom_op.py +++ b/python/sglang/srt/custom_op.py @@ -20,7 +20,7 @@ class CustomOp(nn.Module): raise NotImplementedError def forward_hip(self, *args, **kwargs): - return self.forward_native(*args, **kwargs) + return self.forward_cuda(*args, **kwargs) def forward_xpu(self, *args, **kwargs): return self.forward_native(*args, **kwargs)