remove unnecessary check

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