Skip to content

Commit

Permalink
Use forward_cuda to execute custom op for hip platform (#3305)
Browse files Browse the repository at this point in the history
Co-authored-by: wunhuang <[email protected]>
  • Loading branch information
kkHuang-amd and wunhuang authored Feb 5, 2025
1 parent c2723a4 commit 4885b90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/sglang/srt/custom_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ def forward_cuda(self, *args, **kwargs):
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)
Expand Down

0 comments on commit 4885b90

Please sign in to comment.