Skip to content

Commit

Permalink
[fbsync] Skip MPS tests (#8474)
Browse files Browse the repository at this point in the history
Reviewed By: vmoens

Differential Revision: D58283848

fbshipit-source-id: cfc6b13751cbcb99710e385cfa0e0e634b7cb699
  • Loading branch information
NicolasHug authored and facebook-github-bot committed Jun 7, 2024
1 parent 444b318 commit c4d1728
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit c4d1728

Please sign in to comment.