[GraphBolt][CUDA] Minibatch.to() patch. (#7330)

Author: mfbalin

python/dgl/graphbolt/internal/utils.py

@@ -129,6 +129,21 @@ def copy_or_convert_data(
     save_data(data, output_path, output_format)
 
 
+def get_nonproperty_attributes(_obj) -> list:
+    """Get attributes of the class except for the properties."""
+    attributes = [
+        attribute
+        for attribute in dir(_obj)
+        if not attribute.startswith("__")
+        and (
+            not hasattr(type(_obj), attribute)
+            or not isinstance(getattr(type(_obj), attribute), property)
+        )
+        and not callable(getattr(_obj, attribute))
+    ]
+    return attributes
+
+
 def get_attributes(_obj) -> list:
     """Get attributes of the class."""
     attributes = [

python/dgl/graphbolt/minibatch.py

@@ -9,7 +9,7 @@
 from dgl.utils import recursive_apply
 
 from .base import CSCFormatBase, etype_str_to_tuple, expand_indptr
-from .internal import get_attributes
+from .internal import get_attributes, get_nonproperty_attributes
 from .sampled_subgraph import SampledSubgraph
 
 __all__ = ["MiniBatch"]
@@ -556,23 +556,14 @@ def to_pyg_data(self):
     def to(self, device: torch.device):  # pylint: disable=invalid-name
         """Copy `MiniBatch` to the specified device using reflection."""
 
-        def _to(x, device):
+        def _to(x):
             return x.to(device) if hasattr(x, "to") else x
 
-        def apply_to(x, device):
-            return recursive_apply(x, lambda x: _to(x, device))
-
-        transfer_attrs = get_attributes(self)
+        transfer_attrs = get_nonproperty_attributes(self)
 
         for attr in transfer_attrs:
             # Only copy member variables.
-            try:
-                # For read-only attributes such as blocks , setattr will throw
-                # an AttributeError. We catch these exceptions and skip those
-                # attributes.
-                setattr(self, attr, apply_to(getattr(self, attr), device))
-            except AttributeError:
-                continue
+            setattr(self, attr, recursive_apply(getattr(self, attr), _to))
 
         return self