Skip to content
Snippets Groups Projects
Commit fed06dd0 authored by Bryan O'Sullivan's avatar Bryan O'Sullivan
Browse files

worker: count the number of CPUs

This works on the major platforms, and falls back to a safe guess of
1 elsewhere.
parent 4b5d37ca
No related branches found
No related tags found
No related merge requests found
# worker.py - master-slave parallelism support
#
# Copyright 2013 Facebook, Inc.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
import os
def countcpus():
'''try to count the number of CPUs on the system'''
# posix
try:
n = int(os.sysconf('SC_NPROCESSORS_ONLN'))
if n > 0:
return n
except (AttributeError, ValueError):
pass
# windows
try:
n = int(os.environ['NUMBER_OF_PROCESSORS'])
if n > 0:
return n
except (KeyError, ValueError):
pass
return 1
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment