Merge remote-tracking branch 'upstream/dev' into 8587-test-erros-on-pytorch-release-2508-on-series-50

Author: garciadias

monai/apps/auto3dseg/auto_runner.py

@@ -229,7 +229,7 @@ def __init__(
             input = os.path.join(os.path.abspath(work_dir), "input.yaml")
             logger.info(f"Input config is not provided, using the default {input}")
 
-        self.data_src_cfg = dict()
+        self.data_src_cfg = {}
         if isinstance(input, dict):
             self.data_src_cfg = input
         elif isinstance(input, str) and os.path.isfile(input):

monai/auto3dseg/analyzer.py

@@ -15,7 +15,7 @@
 from abc import ABC, abstractmethod
 from collections.abc import Hashable, Mapping
 from copy import deepcopy
-from typing import Any
+from typing import Any, cast
 
 import numpy as np
 import torch
@@ -470,6 +470,7 @@ def __call__(self, data: Mapping[Hashable, MetaTensor]) -> dict[Hashable, MetaTe
         start = time.time()
         image_tensor = d[self.image_key]
         label_tensor = d[self.label_key]
+        # Check if either tensor is on CUDA to determine if we should move both to CUDA for processing
         using_cuda = any(
             isinstance(t, (torch.Tensor, MetaTensor)) and t.device.type == "cuda" for t in (image_tensor, label_tensor)
         )
@@ -480,7 +481,13 @@ def __call__(self, data: Mapping[Hashable, MetaTensor]) -> dict[Hashable, MetaTe
             label_tensor, (MetaTensor, torch.Tensor)
         ):
             if label_tensor.device != image_tensor.device:
-                label_tensor = label_tensor.to(image_tensor.device)  # type: ignore
+                if using_cuda:
+                    # Move both tensors to CUDA when mixing devices
+                    cuda_device = image_tensor.device if image_tensor.device.type == "cuda" else label_tensor.device
+                    image_tensor = cast(MetaTensor, image_tensor.to(cuda_device))
+                    label_tensor = cast(MetaTensor, label_tensor.to(cuda_device))
+                else:
+                    label_tensor = cast(MetaTensor, label_tensor.to(image_tensor.device))
 
         ndas: list[MetaTensor] = [image_tensor[i] for i in range(image_tensor.shape[0])]  # type: ignore
         ndas_label: MetaTensor = label_tensor.astype(torch.int16)  # (H,W,D)

monai/engines/utils.py

@@ -219,8 +219,8 @@ def __call__(
         `kwargs` supports other args for `Tensor.to()` API.
         """
         image, label = default_prepare_batch(batchdata, device, non_blocking, **kwargs)
-        args_ = list()
-        kwargs_ = dict()
+        args_ = []
+        kwargs_ = {}
 
         def _get_data(key: str) -> torch.Tensor:
             data = batchdata[key]

monai/transforms/inverse.py

@@ -282,8 +282,8 @@ def track_transform_meta(
                     msg += f" for key {key}"
 
                 pend = out_obj.pending_operations[-1]
-                statuses = pend.get(TraceKeys.STATUSES, dict())
-                messages = statuses.get(TraceStatusKeys.PENDING_DURING_APPLY, list())
+                statuses = pend.get(TraceKeys.STATUSES, {})
+                messages = statuses.get(TraceStatusKeys.PENDING_DURING_APPLY, [])
                 messages.append(msg)
                 statuses[TraceStatusKeys.PENDING_DURING_APPLY] = messages
                 info[TraceKeys.STATUSES] = statuses

tests/apps/test_auto3dseg.py

@@ -393,6 +393,7 @@ def test_label_stats_mixed_device_analyzer(self, input_params):
         result = analyzer({"image": image_tensor, "label": label_tensor})
         report = result["label_stats"]
 
+        # Verify report format and computation succeeded despite mixed/unified devices
         assert verify_report_format(report, analyzer.get_report_format())
         assert report[LabelStatsKeys.LABEL_UID] == [0, 1]
 

tests/metrics/test_ssim_metric.py

@@ -21,7 +21,7 @@
 
 class TestSSIMMetric(unittest.TestCase):
 
-    def test2d_gaussian(self):
+    def test_2d_gaussian(self):
         set_determinism(0)
         preds = torch.abs(torch.randn(2, 3, 16, 16))
         target = torch.abs(torch.randn(2, 3, 16, 16))
@@ -32,9 +32,9 @@ def test2d_gaussian(self):
         metric(preds, target)
         result = metric.aggregate()
         expected_value = 0.045415
-        self.assertTrue(expected_value - result.item() < 0.000001)
+        self.assertTrue(abs(expected_value - result.item()) < 0.000001)
 
-    def test2d_uniform(self):
+    def test_2d_uniform(self):
         set_determinism(0)
         preds = torch.abs(torch.randn(2, 3, 16, 16))
         target = torch.abs(torch.randn(2, 3, 16, 16))
@@ -45,9 +45,9 @@ def test2d_uniform(self):
         metric(preds, target)
         result = metric.aggregate()
         expected_value = 0.050103
-        self.assertTrue(expected_value - result.item() < 0.000001)
+        self.assertTrue(abs(expected_value - result.item()) < 0.000001)
 
-    def test3d_gaussian(self):
+    def test_3d_gaussian(self):
         set_determinism(0)
         preds = torch.abs(torch.randn(2, 3, 16, 16, 16))
         target = torch.abs(torch.randn(2, 3, 16, 16, 16))