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