brozek/vspursuer
1
0
Fork 0
mirror of https://github.com/Brandon-Rozek/matmod.git synced 2025-07-31 21:01:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Configurable CPU usage

Browse source
This commit is contained in:
Brandon Rozek 2025-02-09 11:05:40 -05:00
parent 2d50cd9479
commit 54dc6e503c
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
vspursuer.py
View file

@ -11,6 +11,7 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description="VSP Checker") parser = argparse.ArgumentParser(description="VSP Checker")
parser.add_argument("--verbose", action='store_true', help="Print out all parsed matrices") 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("-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()) args = vars(parser.parse_args())
data_file_path = args.get("i") data_file_path = args.get("i")
@ -33,7 +34,8 @@ if __name__ == "__main__":
solutions_expanded.append((model, impfunction, mconjunction, mdisjunction)) solutions_expanded.append((model, impfunction, mconjunction, mdisjunction))
num_has_vsp = 0 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 = [ results = [
pool.apply_async(has_vsp, (model, impfunction, mconjunction, mdisjunction,)) pool.apply_async(has_vsp, (model, impfunction, mconjunction, mdisjunction,))
for model, impfunction, mconjunction, mdisjunction in solutions_expanded for model, impfunction, mconjunction, mdisjunction in solutions_expanded