@RUNIF(mps=False)

Author: Borda

tests/tests_fabric/plugins/precision/test_bitsandbytes.py

@@ -28,6 +28,7 @@
 
 
 @pytest.mark.skipif(_BITSANDBYTES_AVAILABLE, reason="bitsandbytes needs to be unavailable")
+@RunIf(mps=False)  # skip on MPS as Bitsandbytes is only supported on CUDA GPUs
 def test_bitsandbytes_plugin(monkeypatch):
     module = lightning.fabric.plugins.precision.bitsandbytes
     monkeypatch.setattr(module, "_BITSANDBYTES_AVAILABLE", lambda: True)

tests/tests_pytorch/plugins/precision/test_bitsandbytes.py

@@ -22,9 +22,11 @@
 from lightning.fabric.plugins.precision.bitsandbytes import _BITSANDBYTES_AVAILABLE
 from lightning.pytorch import LightningModule, Trainer
 from lightning.pytorch.plugins.precision.bitsandbytes import BitsandbytesPrecision
+from tests_pytorch.helpers.runif import RunIf
 
 
 @pytest.mark.skipif(_BITSANDBYTES_AVAILABLE, reason="bitsandbytes needs to be unavailable")
+@RunIf(mps=False)  # skip on MPS as Bitsandbytes is only supported on CUDA GPUs
 def test_bitsandbytes_plugin(monkeypatch):
     module = lightning.fabric.plugins.precision.bitsandbytes
     monkeypatch.setattr(module, "_BITSANDBYTES_AVAILABLE", lambda: True)