From 4eb7c1cbf344067d08ba18e2487cee52d413b995 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Thu, 6 Jun 2024 11:39:04 +0100 Subject: [PATCH] Skip MPS tests --- test/conftest.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/conftest.py b/test/conftest.py index a9768598ded..89b4946e612 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -49,7 +49,8 @@ def pytest_collection_modifyitems(items): # There are special cases though, see below item.add_marker(pytest.mark.skip(reason=CUDA_NOT_AVAILABLE_MSG)) - if needs_mps and not torch.backends.mps.is_available(): + # TODO: uncoment when MPS works again - see FIXME in setup.py + if needs_mps: # and not torch.backends.mps.is_available(): item.add_marker(pytest.mark.skip(reason=MPS_NOT_AVAILABLE_MSG)) if IN_FBCODE: