Improving gpu_cache test. (#6195)

64df37f7 · Andrei Ivanov · GitHub · f6dec359 · 64df37f7
Unverified Commit 64df37f7 authored Aug 22, 2023 by Andrei Ivanov Committed by GitHub Aug 23, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/python/common/cuda/test_gpu_cache.py tests/python/common/cuda/test_gpu_cache.py +1 -1

No files found.
--- a/tests/python/common/cuda/test_gpu_cache.py
+++ b/tests/python/common/cuda/test_gpu_cache.py
@@ -25,7 +25,7 @@ D = 5


 def generate_graph(idtype, grad=False, add_data=True):
-    g = dgl.DGLGraph().to(F.ctx(), dtype=idtype)
+    g = dgl.graph([]).to(F.ctx(), dtype=idtype)
    g.add_nodes(10)
    u, v = [], []
    # create a graph where 0 is the source and 9 is the sink