diff --git a/modules/zluda_hijacks.py b/modules/zluda_hijacks.py
index 642c7530d..ea906e7c0 100644
--- a/modules/zluda_hijacks.py
+++ b/modules/zluda_hijacks.py
@@ -3,12 +3,22 @@ from modules import rocm
 
 
 _topk = torch.topk
-def topk(tensor: torch.Tensor, *args, **kwargs):
-    device = tensor.device
-    values, indices = _topk(tensor.cpu(), *args, **kwargs)
+def topk(input: torch.Tensor, *args, **kwargs): # pylint: disable=redefined-builtin
+    device = input.device
+    values, indices = _topk(input.cpu(), *args, **kwargs)
     return torch.return_types.topk((values.to(device), indices.to(device),))
 
 
+_fft_fftn = torch.fft.fftn
+def fft_fftn(input: torch.Tensor, *args, **kwargs) -> torch.Tensor: # pylint: disable=redefined-builtin
+    return _fft_fftn(input.cpu(), *args, **kwargs).to(input.device)
+
+
+_fft_ifftn = torch.fft.ifftn
+def fft_ifftn(input: torch.Tensor, *args, **kwargs) -> torch.Tensor: # pylint: disable=redefined-builtin
+    return _fft_ifftn(input.cpu(), *args, **kwargs).to(input.device)
+
+
 def jit_script(f, *_, **__): # experiment / provide dummy graph
     f.graph = torch._C.Graph() # pylint: disable=protected-access
     return f
@@ -17,4 +27,6 @@ def jit_script(f, *_, **__): # experiment / provide dummy graph
 def do_hijack():
     torch.version.hip = rocm.version
     torch.topk = topk
+    torch.fft.fftn = fft_fftn
+    torch.fft.ifftn = fft_ifftn
     torch.jit.script = jit_script