2020 )
2121)
2222@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
23- @testing .gpu
2423class TestFft (unittest .TestCase ):
2524 @testing .for_all_dtypes ()
2625 @testing .numpy_cupy_allclose (
@@ -77,7 +76,6 @@ def test_ifft(self, xp, dtype):
7776 {"shape" : (3 , 4 ), "s" : (1 , 0 ), "axes" : None , "norm" : None },
7877)
7978@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
80- @testing .gpu
8179class TestFft2 (unittest .TestCase ):
8280 @testing .for_all_dtypes ()
8381 @testing .numpy_cupy_allclose (
@@ -135,7 +133,6 @@ def test_ifft2(self, xp, dtype):
135133 {"shape" : (0 , 0 , 5 ), "s" : None , "axes" : None , "norm" : None },
136134)
137135@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
138- @testing .gpu
139136class TestFftn (unittest .TestCase ):
140137 @testing .for_all_dtypes ()
141138 @testing .numpy_cupy_allclose (
@@ -176,7 +173,6 @@ def test_ifftn(self, xp, dtype):
176173 )
177174)
178175@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
179- @testing .gpu
180176class TestRfft (unittest .TestCase ):
181177 @testing .for_all_dtypes (no_complex = True )
182178 @testing .numpy_cupy_allclose (
@@ -208,7 +204,6 @@ def test_irfft(self, xp, dtype):
208204 {"shape" : (2 , 3 , 4 ), "s" : None , "axes" : (), "norm" : None },
209205)
210206@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
211- @testing .gpu
212207class TestRfft2EmptyAxes (unittest .TestCase ):
213208 @testing .for_all_dtypes (no_complex = True )
214209 def test_rfft2 (self , dtype ):
@@ -230,7 +225,6 @@ def test_irfft2(self, dtype):
230225 {"shape" : (2 , 3 , 4 ), "s" : None , "axes" : (), "norm" : None },
231226)
232227@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
233- @testing .gpu
234228class TestRfftnEmptyAxes (unittest .TestCase ):
235229 @testing .for_all_dtypes (no_complex = True )
236230 def test_rfftn (self , dtype ):
@@ -257,7 +251,6 @@ def test_irfftn(self, dtype):
257251 )
258252)
259253@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
260- @testing .gpu
261254class TestHfft (unittest .TestCase ):
262255 @testing .for_all_dtypes ()
263256 @testing .numpy_cupy_allclose (
@@ -290,7 +283,6 @@ def test_ihfft(self, xp, dtype):
290283 {"n" : 100 , "d" : 2 },
291284)
292285@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
293- @testing .gpu
294286class TestFftfreq (unittest .TestCase ):
295287 @testing .for_all_dtypes ()
296288 @testing .numpy_cupy_allclose (
@@ -325,7 +317,6 @@ def test_rfftfreq(self, xp, dtype):
325317 {"shape" : (10 , 10 ), "axes" : (0 , 1 )},
326318)
327319@pytest .mark .usefixtures ("allow_fall_back_on_numpy" )
328- @testing .gpu
329320class TestFftshift (unittest .TestCase ):
330321 @testing .for_all_dtypes ()
331322 @testing .numpy_cupy_allclose (
0 commit comments