0c156573a383620af5194c485bdfe642ea5b33e5,benchmarks/benchmarks/api/bench_batch.py,,track_time,#Any#,9

Before Change

    // prepare graph
    graphs = []
    for i in range(batch_size):
        u = torch.randint(20, (40,))
        v = torch.randint(20, (40,))
        graphs.append(dgl.graph((u, v)).to(device))

    // dry run
    for i in range(10):
        g = dgl.batch(graphs)

    // timing

After Change

@utils.parametrize("batch_size", [4, 32, 256, 1024])
def track_time(batch_size):
    device = utils.get_bench_device()
    ds = dgl.data.QM7bDataset()
    // prepare graph
    graphs = []
    for graph in ds[0:batch_size][0]:
        g = graph.to(device)
        graphs.append(g)

    // dry run
    for i in range(10):
        g = dgl.batch(graphs)

    // timing

Instances