Merge branch 'branch-23.12' into generators

rapidsai · Oct 30, 2023 · bfed89f · bfed89f
2 parents 36ffccf + e0c3a93
commit bfed89f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/cugraph-dgl/examples/graphsage/node-classification.py b/python/cugraph-dgl/examples/graphsage/node-classification.py
@@ -243,7 +243,9 @@ def train(args, device, g, dataset, model):
 
     else:
         g = g.to("cuda" if args.mode == "gpu_dgl" else "cpu")
-    device = torch.device("cpu" if args.mode == "cpu" else "cuda")
+    device = torch.device(
+        "cpu" if args.mode == "cpu" or args.mode == "mixed" else "cuda"
+    )
 
     # create GraphSAGE model
     feat_shape = (