[GraphBolt][CUDA] Temporarily disable overlap for the multiGPU example. (#7009)

0a2f40f6 · Muhammed Fatih BALIN · GitHub · 3a79f021 · 0a2f40f6
Unverified Commit 0a2f40f6 authored Jan 25, 2024 by Muhammed Fatih BALIN Committed by GitHub Jan 25, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

examples/multigpu/graphbolt/node_classification.py examples/multigpu/graphbolt/node_classification.py +4 -1

No files found.
--- a/examples/multigpu/graphbolt/node_classification.py
+++ b/examples/multigpu/graphbolt/node_classification.py
@@ -141,7 +141,10 @@ def create_dataloader(
    if args.storage_device == "cpu":
        datapipe = datapipe.copy_to(device)

-    dataloader = gb.DataLoader(datapipe, args.num_workers)
+    # Until https://github.com/dmlc/dgl/issues/7008, overlap should be False.
+    dataloader = gb.DataLoader(
+        datapipe, args.num_workers, overlap_feature_fetch=False
+    )

    # Return the fully-initialized DataLoader object.
    return dataloader