[Misc] DGL multi-gpu example run error work around (#6999)

examples/multigpu/node_classification_sage.py

@@ -171,11 +171,15 @@ def train(
     use_uva,
 ):
     # Instantiate a neighbor sampler
+    if args.mode == "benchmark":
+        # A work-around to prevent CUDA running error. For more details, please
+        # see https://github.com/dmlc/dgl/issues/6697.
+        sampler = NeighborSampler([10, 10, 10], fused=False)
+    else:
         sampler = NeighborSampler(
             [10, 10, 10],
             prefetch_node_feats=["feat"],
             prefetch_labels=["label"],
-        fused=(args.mode != "benchmark"),
         )
     train_dataloader = DataLoader(
         g,