EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: wunhuang <wunhuang@amd.com>
This commit is contained in:
kk
2025-02-05 10:58:17 +08:00
committed by GitHub
parent c2723a42a5
commit 4885b90802
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/custom_op.py
View File

@@ -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)