Unverified Commit 64df37f7 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

Improving gpu_cache test. (#6195)

parent f6dec359
...@@ -25,7 +25,7 @@ D = 5 ...@@ -25,7 +25,7 @@ D = 5
def generate_graph(idtype, grad=False, add_data=True): 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) g.add_nodes(10)
u, v = [], [] u, v = [], []
# create a graph where 0 is the source and 9 is the sink # create a graph where 0 is the source and 9 is the sink
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment