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

Co-authored-by: wunhuang <wunhuang@amd.com>
2025-02-05 02:58:17 +00:00
parent c2723a42a5
commit 4885b90802
1 changed files with 1 additions and 1 deletions
@@ -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)