Activation of test_exclude_seed_edges_gpu on CPU.

dmlc · Jul 26, 2024 · 6267f4c · 6267f4c
1 parent 8473bee
commit 6267f4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/pytorch/graphbolt/test_utils.py b/tests/python/pytorch/graphbolt/test_utils.py
@@ -117,7 +117,7 @@ def test_exclude_seed_edges_homo_cpu():
             )
 
 
-def test_exclude_seed_edges_gpu():
+def test_exclude_seed_edges():
     graph = dgl.graph(([5, 0, 7, 7, 2, 4], [0, 1, 2, 2, 3, 4]))
     graph = gb.from_dglgraph(graph, is_homogeneous=True).to(F.ctx())
     items = torch.LongTensor([[0, 3], [4, 4]])