Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "ab2bd1f13dd982be501b5874e9d4eb2217b068bc"
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 ...@@ -7,10 +7,9 @@ import dgl.function as fn
from .. import utils from .. import utils
@utils.skip_if_gpu()
@utils.benchmark('time') @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('format', ['coo', 'csc'])
@utils.parametrize('seed_nodes_num', [200, 5000, 20000]) @utils.parametrize('seed_nodes_num', [200, 5000, 20000])
@utils.parametrize('fanout', [5, 20, 40]) @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