From 54dc6e503ce383cbc7f32d21dcade04b144111d0 Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Sun, 9 Feb 2025 11:05:40 -0500 Subject: [PATCH] Configurable CPU usage --- vspursuer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vspursuer.py b/vspursuer.py index 66716a3..449cfa0 100755 --- a/vspursuer.py +++ b/vspursuer.py @@ -11,6 +11,7 @@ if __name__ == "__main__": parser = argparse.ArgumentParser(description="VSP Checker") parser.add_argument("--verbose", action='store_true', help="Print out all parsed matrices") parser.add_argument("-i", type=str, help="Path to MaGIC ugly data file") + parser.add_argument("-c", type=int, help="Number of CPUs to use. Default: MAX - 2.") args = vars(parser.parse_args()) data_file_path = args.get("i") @@ -33,7 +34,8 @@ if __name__ == "__main__": solutions_expanded.append((model, impfunction, mconjunction, mdisjunction)) num_has_vsp = 0 - with mp.Pool(processes=max(cpu_count() - 2, 1)) as pool: + num_cpu = args.get("c", max(cpu_count() - 2, 1)) + with mp.Pool(processes=num_cpu) as pool: results = [ pool.apply_async(has_vsp, (model, impfunction, mconjunction, mdisjunction,)) for model, impfunction, mconjunction, mdisjunction in solutions_expanded