mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-06-10 16:28:19 +00:00
19 lines
341 B
Python
19 lines
341 B
Python
import sys
|
|
import time
|
|
|
|
import cuda.nvbench as nvbench
|
|
|
|
|
|
def throughput_bench(state: nvbench.State) -> None:
|
|
def launcher(launch: nvbench.Launch):
|
|
time.sleep(1)
|
|
|
|
state.exec(launcher)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
b = nvbench.register(throughput_bench)
|
|
b.set_is_cpu_only(True)
|
|
|
|
nvbench.run_all_benchmarks(sys.argv)
|