fix the bug

dmlc · Jan 17, 2024 · 93c8a18 · 93c8a18
1 parent abf466c
commit 93c8a18
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/dgl/graphbolt/impl/fused_csc_sampling_graph.py b/python/dgl/graphbolt/impl/fused_csc_sampling_graph.py
@@ -990,7 +990,8 @@ def _pin(x):
             if hasattr(x, "pin_memory_"):
                 x.pin_memory_()
             elif (
-                not x.is_pinned()
+                isinstance(x, torch.Tensor)
+                and not x.is_pinned()
                 and x.device.type == "cpu"
                 and x.is_contiguous()
             ):