[MPS] Add regression test for fft.fftfreq (#137215)

pytorchbot · hvaara · web-flow · commit 783a6a424c30 · 2024-10-02T15:19:35.000-07:00
[MPS] Add regression test for `fft.fftfreq` (#135440) The issue reported in #135223 was already solved in #128393. This PR adds a regression test for it. Fixes #135223 Pull Request resolved: #135440 Approved by: https://github.com/ezyang (cherry picked from commit 09287e3) Co-authored-by: Roy Hvaara <roy@lightyear.no>
diff --git a/test/test_mps.py b/test/test_mps.py
@@ -2732,6 +2732,12 @@ def test_ifft(self):
         # Expecting the inverted to yield the original signal
         self.assertEqual(ifft_result, signal)
 
+    # Regression test for https://github.com/pytorch/pytorch/issues/135223
+    def test_fftfreq(self):
+        freq_cpu = torch.fft.fftfreq(10**4, device='cpu')
+        freq_mps = torch.fft.fftfreq(10**4, device='mps')
+        self.assertEqual(freq_cpu, freq_mps)
+
     def test_instance_norm(self):
         def helper(shape, eps=1, momentum=0.1, wts=False, channels_last=False, track_running_stats=True, test_module=False):
 
