"src/vscode:/vscode.git/clone" did not exist on "77c80489f3ab6cf4ac208d2d317ebe0bc694e785"
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
......@@ -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)
......
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