1414 get_all_dtypes ,
1515 get_complex_dtypes ,
1616 get_float_dtypes ,
17- is_cuda_device ,
1817)
1918
2019
@@ -443,11 +442,6 @@ def setup_method(self):
443442 @pytest .mark .parametrize ("norm" , [None , "backward" , "forward" , "ortho" ])
444443 @pytest .mark .parametrize ("order" , ["C" , "F" ])
445444 def test_fftn (self , dtype , axes , norm , order ):
446- if is_cuda_device ():
447- if order == "C" and axes == (0 , 1 , 2 ):
448- pass
449- else :
450- pytest .skip ("SAT-7587" )
451445 a_np = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype , order )
452446 a = dpnp .array (a_np )
453447
@@ -482,9 +476,6 @@ def test_fftn_repeated_axes(self, axes):
482476 @pytest .mark .parametrize ("axes" , [(2 , 3 , 3 , 2 ), (0 , 0 , 3 , 3 )])
483477 @pytest .mark .parametrize ("s" , [(5 , 4 , 3 , 3 ), (7 , 8 , 10 , 9 )])
484478 def test_fftn_repeated_axes_with_s (self , axes , s ):
485- if is_cuda_device ():
486- if axes == (0 , 0 , 3 , 3 ) and s == (7 , 8 , 10 , 9 ):
487- pytest .skip ("SAT-7587" )
488479 a_np = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype = numpy .complex64 )
489480 a = dpnp .array (a_np )
490481
@@ -504,11 +495,6 @@ def test_fftn_repeated_axes_with_s(self, axes, s):
504495 @pytest .mark .parametrize ("axes" , [(0 , 1 , 2 , 3 ), (1 , 2 , 1 , 2 ), (2 , 2 , 2 , 3 )])
505496 @pytest .mark .parametrize ("s" , [(2 , 3 , 4 , 5 ), (5 , 4 , 7 , 8 ), (2 , 5 , 1 , 2 )])
506497 def test_fftn_out (self , axes , s ):
507- if is_cuda_device ():
508- if axes == (0 , 1 , 2 , 3 ):
509- pytest .skip ("SAT-7587" )
510- elif s == (2 , 5 , 1 , 2 ) and axes in [(1 , 2 , 1 , 2 ), (2 , 2 , 2 , 3 )]:
511- pytest .skip ("SAT-7587" )
512498 a_np = generate_random_numpy_array ((2 , 3 , 4 , 5 ), dtype = numpy .complex64 )
513499 a = dpnp .array (a_np )
514500
@@ -1082,9 +1068,6 @@ def test_rfftn_repeated_axes_with_s(self, axes, s):
10821068 @pytest .mark .parametrize ("axes" , [(0 , 1 , 2 , 3 ), (1 , 2 , 1 , 2 ), (2 , 2 , 2 , 3 )])
10831069 @pytest .mark .parametrize ("s" , [(2 , 3 , 4 , 5 ), (5 , 6 , 7 , 9 ), (2 , 5 , 1 , 2 )])
10841070 def test_rfftn_out (self , axes , s ):
1085- if is_cuda_device ():
1086- if axes == (0 , 1 , 2 , 3 ) and s == (2 , 5 , 1 , 2 ):
1087- pytest .skip ("SAT-7587" )
10881071 x = numpy .random .uniform (- 10 , 10 , 120 )
10891072 a_np = numpy .array (x , dtype = numpy .float32 ).reshape (2 , 3 , 4 , 5 )
10901073 a = dpnp .asarray (a_np )
0 commit comments