[BugFix] fetch device from graph (#4245)

Co-authored-by: Xin Yao <xiny@nvidia.com>

[BugFix] fetch device from graph (#4245)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2f322a94 · Rhett Ying · GitHub · e26d2064 · 2f322a94
Unverified Commit 2f322a94 authored Jul 14, 2022 by Rhett Ying Committed by GitHub Jul 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/dgl/dataloading/dataloader.py python/dgl/dataloading/dataloader.py +1 -1

No files found.
--- a/python/dgl/dataloading/dataloader.py
+++ b/python/dgl/dataloading/dataloader.py
@@ -941,7 +941,7 @@ class EdgeDataLoader(DataLoader):
            if use_uva:
                device = torch.cuda.current_device()
            else:
-                device = self.graph.device
+                device = graph.device
        device = _get_device(device)

        if isinstance(graph_sampler, BlockSampler):