@@ -340,8 +340,8 @@ def tmp(t):
340
340
('kthvalue' , small_3d_unique , lambda t : [3 , - 1 ], 'neg_dim' ),
341
341
('lerp' , small_3d , lambda t : [small_3d (t ), 0.3 ]),
342
342
('max' , small_3d_unique , lambda t : []),
343
- ('max' , small_3d_unique , lambda t : [1 ], 'dim' , types , False , skipIfRocm ),
344
- ('max' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' , types , False , skipIfRocm ),
343
+ ('max' , small_3d_unique , lambda t : [1 ], 'dim' ),
344
+ ('max' , small_3d_unique , lambda t : [- 1 ], 'neg_dim' ),
345
345
('max' , medium_2d , lambda t : [medium_2d (t )], 'elementwise' ),
346
346
('min' , small_3d_unique , lambda t : []),
347
347
('min' , small_3d_unique , lambda t : [1 ], 'dim' , types , False , skipIfRocm ),
@@ -350,9 +350,9 @@ def tmp(t):
350
350
('mean' , small_3d , lambda t : []),
351
351
('mean' , small_3d , lambda t : [- 1 ], 'neg_dim' ),
352
352
('mean' , small_3d , lambda t : [1 ], 'dim' ),
353
- ('mode' , small_3d , lambda t : [], '' , types , False , skipIfRocm ),
354
- ('mode' , small_3d , lambda t : [1 ], 'dim' , types , False , skipIfRocm ),
355
- ('mode' , small_3d , lambda t : [- 1 ], 'neg_dim' , types , False , skipIfRocm ),
353
+ ('mode' , small_3d , lambda t : []),
354
+ ('mode' , small_3d , lambda t : [1 ], 'dim' ),
355
+ ('mode' , small_3d , lambda t : [- 1 ], 'neg_dim' ),
356
356
('mvlgamma' , lambda t : tensor_clamp (small_2d (t ), 0.1 , 10 ), lambda t : [1 ], '2d_p=1' , float_types_no_half ),
357
357
('mvlgamma' , lambda t : tensor_clamp (small_2d (t ), 0.6 , 10 ), lambda t : [2 ], '2d_p=2' , float_types_no_half ),
358
358
('remainder' , small_3d , lambda t : [3 ], 'value' , types , False , "skipIfRocm:HalfTensor" ),
@@ -957,7 +957,6 @@ def test_broadcast_cpu(self):
957
957
def test_broadcast_gpu (self ):
958
958
self ._test_broadcast (torch .randn (5 , 5 ).cuda ())
959
959
960
- @skipIfRocm
961
960
def test_min_max_nan (self ):
962
961
tests = [(lambda x : x .min (), 'min' ),
963
962
(lambda x : x .max (), 'max' ),
@@ -1723,7 +1722,6 @@ def test_tensor_scatterAdd(self):
1723
1722
def test_tensor_scatterFill (self ):
1724
1723
TestTorch ._test_scatter_base (self , lambda t : t .cuda (), 'scatter_' , True , test_bounds = False )
1725
1724
1726
- @skipIfRocm
1727
1725
def test_min_max_inits (self ):
1728
1726
# Testing if THC_reduceAll received the correct index initialization.
1729
1727
# This affects the result of THC_reduceAll operations at extreme values
@@ -1737,11 +1735,9 @@ def test_min_max_inits(self):
1737
1735
_ , v = y .min (dim = 0 )
1738
1736
self .assertEqual (v , expected )
1739
1737
1740
- @skipIfRocm
1741
1738
def test_max_with_inf (self ):
1742
1739
TestTorch ._test_max_with_inf (self , (torch .half , torch .float , torch .double ), 'cuda' )
1743
1740
1744
- @skipIfRocm
1745
1741
def test_min_with_inf (self ):
1746
1742
TestTorch ._test_min_with_inf (self , (torch .half , torch .float , torch .double ), 'cuda' )
1747
1743
0 commit comments