diff --git a/python/cuda/bench/__init__.py b/python/cuda/bench/__init__.py index 119f994b..f197c539 100644 --- a/python/cuda/bench/__init__.py +++ b/python/cuda/bench/__init__.py @@ -134,11 +134,6 @@ def __dir__(): ) -__doc__ = """ -CUDA Kernel Benchmarking Library Python API -""" - - def _get_register(): if _register is not None: return _register diff --git a/python/cuda/bench/_decorators.py b/python/cuda/bench/_decorators.py index ce15ded0..dd55b209 100644 --- a/python/cuda/bench/_decorators.py +++ b/python/cuda/bench/_decorators.py @@ -98,6 +98,9 @@ def register(fn=None, /): Example ------- + + .. code-block:: python + @bench.register() @bench.axis.int64("Elements", [1024, 2048]) @bench.option.min_samples(10)