Skip to content

Commit 819bcc1

Browse files
committed
Add RAYON_NUM_THREADS environment variable support to benchmark output
1 parent 73c895d commit 819bcc1

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

benchmarks/collect_gil_bench.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,8 +290,10 @@ def run(
290290
)
291291
args = parser.parse_args()
292292

293+
import os
294+
rayon_threads = os.environ.get('RAYON_NUM_THREADS', 'default')
293295
print(f"\n\nRunning benchmark with {args.batches} batches, {args.batch_size} rows per batch")
294-
print(f"Partitions: {args.partitions or args.batches}, Workload: {args.workload}")
296+
print(f"Partitions: {args.partitions or args.batches}, Workload: {args.workload}, RAYON_NUM_THREADS: {rayon_threads}")
295297
print("-" * 60)
296298

297299
run(

0 commit comments

Comments
 (0)