zluda hijack torch jit

Author: lshqqytiger

modules/zluda_hijacks.py

@@ -8,6 +8,12 @@ def topk(tensor: torch.Tensor, *args, **kwargs):
     return torch.return_types.topk((values.to(device), indices.to(device),))
 
 
+def jit_script(f, *_, **__): # experiment / provide dummy graph
+    f.graph = torch._C.Graph() # pylint: disable=protected-access
+    return f
+
+
 def do_hijack():
     torch.version.hip = "5.7"
     torch.topk = topk
+    torch.jit.script = jit_script