Skip to content

Commit 88cb632

Browse files
Add examples/cpu_only.py based on code from PR feedback
#237 (comment)
1 parent 17ac4e6 commit 88cb632

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

python/examples/cpu_only.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import sys
2+
import time
3+
4+
import cuda.nvbench as nvbench
5+
6+
7+
def throughput_bench(state: nvbench.State) -> None:
8+
def launcher(launch: nvbench.Launch):
9+
time.sleep(1)
10+
11+
state.exec(launcher)
12+
13+
14+
if __name__ == "__main__":
15+
b = nvbench.register(throughput_bench)
16+
b.setIsCPUOnly(True)
17+
18+
nvbench.run_all_benchmarks(sys.argv)

0 commit comments

Comments
 (0)