mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-03-26 01:57:24 +00:00
807917cf54
Add sphinx-combined folder that builds combined C++ & Python docs Fixed relative text alignment in docstrings to fix autodoc warnigns Renamed cuda.bench.test_cpp_exception and cuda.bench.test_py_exception functions to start with underscore, signaling that these functions are internal and should not be documented Account for test_cpp_exceptions -> _test_cpp_exception, same for *_py_* Fix cpp_benchmarks, add py_benchmarks 1. Fixed xrefs in docs/sphinx-combined/cpp_benchmarks.md, which is built on top of docs/benchmarks.md Added level-1 heading, and pushed existing headings one level down. 2. Added py_benchmarks.md to document benchmarking of Python scripts. 3. Rearranged entries in index.rst so that overview documents come before API enumeration. Make sure to reset __module__ of reexported symbols to be cuda.bench Enumerate free functions in nvbench:: namespace Tweak to index.rst intro sentence and title Changed title, fixed references, added intro borrowed from README Fix punctuation in one of the itemlist item text Hide TOC from the index page. It is too long and confusing
91 lines
2.0 KiB
Python
91 lines
2.0 KiB
Python
import json
|
|
|
|
import cuda.bench as bench
|
|
import pytest
|
|
|
|
|
|
def test_cpp_exception():
|
|
with pytest.raises(RuntimeError, match="Test"):
|
|
bench._nvbench._test_cpp_exception()
|
|
|
|
|
|
def test_py_exception():
|
|
with pytest.raises(bench.NVBenchRuntimeError, match="Test"):
|
|
bench._nvbench._test_py_exception()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"cls", [bench.CudaStream, bench.State, bench.Launch, bench.Benchmark]
|
|
)
|
|
def test_api_ctor(cls):
|
|
with pytest.raises(TypeError, match="No constructor defined!"):
|
|
cls()
|
|
|
|
|
|
def t_bench(state: bench.State):
|
|
s = {"a": 1, "b": 0.5, "c": "test", "d": {"a": 1}}
|
|
|
|
def launcher(launch: bench.Launch):
|
|
for _ in range(10000):
|
|
_ = json.dumps(s)
|
|
|
|
state.exec(launcher)
|
|
|
|
|
|
def test_cpu_only():
|
|
b = bench.register(t_bench)
|
|
b.set_is_cpu_only(True)
|
|
|
|
bench.run_all_benchmarks(["-q", "--profile"])
|
|
|
|
|
|
def docstring_check(doc_str: str) -> None:
|
|
assert isinstance(doc_str, str)
|
|
assert len(doc_str) > 0
|
|
|
|
|
|
def obj_has_docstring_check(o: object) -> None:
|
|
docstring_check(o.__doc__)
|
|
|
|
|
|
def test_module_doc():
|
|
obj_has_docstring_check(bench)
|
|
|
|
|
|
def test_register_doc():
|
|
obj_has_docstring_check(bench.register)
|
|
|
|
|
|
def test_run_all_benchmarks_doc():
|
|
obj_has_docstring_check(bench.run_all_benchmarks)
|
|
|
|
|
|
def test_State_doc():
|
|
cl = bench.State
|
|
obj_has_docstring_check(cl)
|
|
obj_has_docstring_check(cl.exec)
|
|
obj_has_docstring_check(cl.get_int64)
|
|
obj_has_docstring_check(cl.get_float64)
|
|
obj_has_docstring_check(cl.get_string)
|
|
obj_has_docstring_check(cl.skip)
|
|
|
|
|
|
def test_Launch_doc():
|
|
cl = bench.Launch
|
|
obj_has_docstring_check(cl)
|
|
obj_has_docstring_check(cl.get_stream)
|
|
|
|
|
|
def test_CudaStream_doc():
|
|
cl = bench.CudaStream
|
|
obj_has_docstring_check(cl)
|
|
|
|
|
|
def test_Benchmark_doc():
|
|
cl = bench.Benchmark
|
|
obj_has_docstring_check(cl)
|
|
obj_has_docstring_check(cl.add_int64_axis)
|
|
obj_has_docstring_check(cl.add_int64_power_of_two_axis)
|
|
obj_has_docstring_check(cl.add_float64_axis)
|
|
obj_has_docstring_check(cl.add_string_axis)
|