From 954d1306c05b2d94a3993fec0e7ae33438129f40 Mon Sep 17 00:00:00 2001 From: Sadra Barikbin Date: Sun, 29 Sep 2024 23:45:14 +0330 Subject: [PATCH] Skip MPS on test_integraion as well --- tests/ignite/metrics/vision/test_object_detection_map.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/ignite/metrics/vision/test_object_detection_map.py b/tests/ignite/metrics/vision/test_object_detection_map.py index de41d317e56..712b2fdebdf 100644 --- a/tests/ignite/metrics/vision/test_object_detection_map.py +++ b/tests/ignite/metrics/vision/test_object_detection_map.py @@ -923,13 +923,16 @@ def test_compute(sample): def test_integration(sample): bs = 3 + device = idist.device() + if device == torch.device("mps"): + pytest.skip("Due to MPS backend out of memory") + def update(engine, i): b = slice(i * bs, (i + 1) * bs) return sample.data[0][b], sample.data[1][b] engine = Engine(update) - device = idist.device() metric_device = "cpu" if device.type == "xla" else device metric_50_95 = ObjectDetectionAvgPrecisionRecall(num_classes=91, device=metric_device) metric_50_95.attach(engine, name="mAP[50-95]")