Skip to content

Commit

Permalink
Merge pull request #17642 from apaszke:fix-no-teardown
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 566314866
  • Loading branch information
jax authors committed Sep 18, 2023
2 parents 32ee27b + 9fa96f8 commit ee846fa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/lobpcg_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -368,10 +368,10 @@ def checkApproxEigs(self, example_name, dtype):
class F32LobpcgTest(LobpcgTest):

def setUp(self):
super().setUp()
# TODO(phawkins): investigate this failure
if jtu.device_under_test() == "gpu":
raise unittest.SkipTest("Test is failing on CUDA gpus")
super().setUp()

def testLobpcgValidatesArguments(self):
A, _ = _concrete_generators(np.float32)['id'](100, 10)
Expand Down Expand Up @@ -408,10 +408,10 @@ def testCallableMatricesF32(self, matrix_name):
class F64LobpcgTest(LobpcgTest):

def setUp(self):
super().setUp()
# TODO(phawkins): investigate this failure
if jtu.device_under_test() == "gpu":
raise unittest.SkipTest("Test is failing on CUDA gpus")
super().setUp()

@parameterized.named_parameters(_make_concrete_cases(f64=True))
@jtu.skip_on_devices("tpu", "iree", "gpu")
Expand Down

0 comments on commit ee846fa

Please sign in to comment.