From 04dcc127079c0079ed3c9e628cd52c1f486b3c14 Mon Sep 17 00:00:00 2001
From: Thomas Skovlund Hansen <thomasskovlundhansen@gmail.com>
Date: Wed, 3 May 2023 13:26:43 +0200
Subject: [PATCH] Add benchmark parameters N_THREADS and N_PARALLEL

---
 Programs/Source/benchmark_priority_queue.mpc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Programs/Source/benchmark_priority_queue.mpc b/Programs/Source/benchmark_priority_queue.mpc
index c848d96d..b621eaf9 100644
--- a/Programs/Source/benchmark_priority_queue.mpc
+++ b/Programs/Source/benchmark_priority_queue.mpc
@@ -33,6 +33,8 @@ INSERT = INSERT or EXTRACT  # Always insert if we are going to extract
 ## General
 KEY_SIZE = lambda n: 32
 VALUE_SIZE = lambda n: log2(n)
+N_THREADS = 1
+N_PARALLEL = 1
 
 ## Insert / ExtractMin
 RANGE = [2**i for i in range(1, 15)]
@@ -49,6 +51,10 @@ POH_PATH_CONSTANT_STASH = True
 ## Sorting
 LENGTHS = [5, 10, 15, 20, 25, 30]
 
+# Set module variables based on parameters
+oram.n_threads = N_THREADS
+oram.n_parallel = N_PARALLEL
+
 # Timing with consecutive ids
 timer_offset = 1000  # Hopefully run timers in an unused range