From 4885b908021768a5220034290edd412c349c925f Mon Sep 17 00:00:00 2001 From: kk <43161300+kkHuang-amd@users.noreply.github.com> Date: Wed, 5 Feb 2025 10:58:17 +0800 Subject: [PATCH] Use forward_cuda to execute custom op for hip platform (#3305) Co-authored-by: wunhuang --- python/sglang/srt/custom_op.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)