@@ -3931,6 +3931,7 @@ def test_is_signed_cuda(self):
3931
3931
self .assertEqual (torch .cuda .HalfTensor (10 ).is_signed (), True )
3932
3932
3933
3933
@skipIfNoLapack
3934
+ @skipIfRocm
3934
3935
def test_gesv (self ):
3935
3936
a = torch .Tensor (((6.80 , - 2.11 , 5.66 , 5.97 , 8.23 ),
3936
3937
(- 6.05 , - 3.30 , 5.36 , - 4.44 , 1.08 ),
@@ -4061,6 +4062,7 @@ def test_gesv_batched_dims(self):
4061
4062
self ._test_gesv_batched_dims (self , lambda t : t )
4062
4063
4063
4064
@skipIfNoLapack
4065
+ @skipIfRocm
4064
4066
def test_qr (self ):
4065
4067
4066
4068
# Since the QR decomposition is unique only up to the signs of the rows of
@@ -4243,10 +4245,12 @@ def _test_trtrs(self, cast):
4243
4245
self .assertEqual (res1 , tb , 0 )
4244
4246
4245
4247
@skipIfNoLapack
4248
+ @skipIfRocm
4246
4249
def test_trtrs (self ):
4247
4250
self ._test_trtrs (self , lambda t : t )
4248
4251
4249
4252
@skipIfNoLapack
4253
+ @skipIfRocm
4250
4254
def test_gels (self ):
4251
4255
def _test_underdetermined (a , b , expectedNorm ):
4252
4256
m = a .size ()[0 ]
@@ -4362,6 +4366,7 @@ def check_norm(a, b, expected_norm, gels_result):
4362
4366
self .assertEqual ((torch .mm (a , tb ) - b ).norm (), expectedNorm , 1e-8 )
4363
4367
4364
4368
@skipIfNoLapack
4369
+ @skipIfRocm
4365
4370
def test_eig (self ):
4366
4371
a = torch .Tensor (((1.96 , 0.00 , 0.00 , 0.00 , 0.00 ),
4367
4372
(- 6.49 , 3.80 , 0.00 , 0.00 , 0.00 ),
0 commit comments