Skip to content

GitLab

Unverified Commit 4885b908 authored by kk's avatar kk Committed by GitHub
Browse files

Use forward_cuda to execute custom op for hip platform (#3305) 


Co-authored-by: default avatarwunhuang <wunhuang@amd.com>
parent c2723a42
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/sglang/srt/custom_op.py
View file @ 4885b908
...@@ -20,7 +20,7 @@ class CustomOp(nn.Module): ...@@ -20,7 +20,7 @@ class CustomOp(nn.Module):
raise NotImplementedError raise NotImplementedError
def forward_hip(self, *args, **kwargs): def forward_hip(self, *args, **kwargs):
return self.forward_native(*args, **kwargs) return self.forward_cuda(*args, **kwargs)
def forward_xpu(self, *args, **kwargs): def forward_xpu(self, *args, **kwargs):
return self.forward_native(*args, **kwargs) return self.forward_native(*args, **kwargs)
......
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