Skip to content

GitLab

"tests/python/common/dataloading/test_dataloader.py" did not exist on "98325b1097877b93dc872727d22ce2f402666e8f"
Unverified Commit 115ac0b9 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

[Benchmark] Enable sample neighbors gpu benchmark (#3060) 

* skip deprecated kernels

* Revert "skip deprecated kernels"

This reverts commit 2ee7f212d11d0858a0f863349a5e02ad04e4bc3c.

* fix
parent 427a5a96
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
benchmarks/benchmarks/api/bench_sample_neighbors.py
View file @ 115ac0b9
......@@ -7,10 +7,9 @@ import dgl.function as fn
from .. import utils
@utils.skip_if_gpu()
@utils.benchmark('time')
@utils.parametrize('graph_name', ['livejournal', 'reddit'])
@utils.parametrize_cpu('graph_name', ['livejournal', 'reddit'])
@utils.parametrize_gpu('graph_name', ['ogbn-arxiv', 'reddit'])
@utils.parametrize('format', ['coo', 'csc'])
@utils.parametrize('seed_nodes_num', [200, 5000, 20000])
@utils.parametrize('fanout', [5, 20, 40])
......
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