brozek/vspursuer
1
0
Fork 0
mirror of https://github.com/Brandon-Rozek/matmod.git synced 2025-11-03 03:11:12 +00:00
Code Issues Projects Releases Packages Wiki Activity

Switched to cpu_count

Browse source
This commit is contained in:
Brandon Rozek 2025-03-19 09:29:02 -04:00
parent c554c53141
commit 154cee0349
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
vspursuer.py
View file

@ -1,5 +1,7 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from os import process_cpu_count
# NOTE: Perhaps we should use process_cpu_count but that's not available to all Python versions
from os import cpu_count
from time import sleep from time import sleep
from typing import Dict, Iterator, Optional, Tuple from typing import Dict, Iterator, Optional, Tuple
import argparse import argparse
@ -64,7 +66,7 @@ if __name__ == "__main__":
num_cpu = args.get("c") num_cpu = args.get("c")
if num_cpu is None: if num_cpu is None:
num_cpu = max(process_cpu_count() - 2, 1) num_cpu = max(cpu_count() - 2, 1)
# Set up parallel verification # Set up parallel verification
num_tested = 0 num_tested = 0
@ -110,6 +112,8 @@ if __name__ == "__main__":
except StopIteration: except StopIteration:
done_parsing = True done_parsing = True
else: else:
# Otherwise the task is still working,
# add it to the next task pool
next_task_pool.append(result_async) next_task_pool.append(result_async)
task_pool = next_task_pool task_pool = next_task_pool
sleep(0.01) sleep(0.01)