Skip to content

Commit b8dc3de

Browse files
Use decorators to register benchmark in RST intro doc
1 parent 0893e3b commit b8dc3de

1 file changed

Lines changed: 5 additions & 9 deletions

File tree

docs/sphinx-combined/py_benchmarks.md

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,14 @@ library to benchmark GPU-aware Python code.
77

88
```python
99
from cuda.bench import State, Launch
10-
from cuda.bench import register, run_all_registered
10+
from cuda.bench import register, axis, options, run_all_registered
1111
from typing import Callable
1212

13-
from my_package import impl
13+
from my_package import impl, generate
1414

15+
@register()
16+
@options.set_name("my_package_kernel")
17+
@axis.add_int64_axis("Elements", [1000, 10000, 100000])
1518
def benchmark_impl(state: State) -> None:
1619

1720
# get state parameters
@@ -29,13 +32,6 @@ def benchmark_impl(state: State) -> None:
2932
state.exec(launch_fn)
3033

3134

32-
bench = register(benchmark_impl)
33-
# provide kernel a name
34-
bench.set_name("my_package_kernel")
35-
# specify default values of parameter to run benchmark with
36-
bench.add_int64_axis("Elements", [1000, 10000, 100000])
37-
38-
3935
if __name__ == "__main__":
4036
import sys
4137
run_all_registered(sys.argv)

0 commit comments

Comments
 (0)