1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bpo-33166: Change os.cpu_count to return active (real) processors (GH-15949)

Browse Source
This commit is contained in:
Steve Dower 2019-09-11 16:15:39 +01:00 committed by Zachary Ware
parent e20134f889
commit aa929273ca
2 changed files with 5 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Misc/NEWS.d/next/Windows/2019-09-11-14-42-04.bpo-36634.8Un8ih.rst Normal file
View File

@ -0,0 +1,2 @@
:func:`os.cpu_count` now returns active processors rather than maximum
processors.

20
Modules/posixmodule.c
View File

@ -12204,23 +12204,9 @@ os_cpu_count_impl(PyObject *module)
{ {
int ncpu = 0; int ncpu = 0;
#ifdef MS_WINDOWS #ifdef MS_WINDOWS
/* Vista is supported and the GetMaximumProcessorCount API is Win7+ /* Declare prototype here to avoid pulling in all of the Win7 APIs in 3.8 */
Need to fallback to Vista behavior if this call isn't present */ DWORD WINAPI GetActiveProcessorCount(WORD group);
HINSTANCE hKernel32; ncpu = GetActiveProcessorCount(ALL_PROCESSOR_GROUPS);
static DWORD(CALLBACK *_GetMaximumProcessorCount)(WORD) = NULL;
Py_BEGIN_ALLOW_THREADS
hKernel32 = GetModuleHandleW(L"KERNEL32");
*(FARPROC*)&_GetMaximumProcessorCount = GetProcAddress(hKernel32,
"GetMaximumProcessorCount");
Py_END_ALLOW_THREADS
if (_GetMaximumProcessorCount != NULL) {
ncpu = _GetMaximumProcessorCount(ALL_PROCESSOR_GROUPS);
}
else {
SYSTEM_INFO sysinfo;
GetSystemInfo(&sysinfo);
ncpu = sysinfo.dwNumberOfProcessors;
}
#elif defined(__hpux) #elif defined(__hpux)
ncpu = mpctl(MPC_GETNUMSPUS, NULL, NULL); ncpu = mpctl(MPC_GETNUMSPUS, NULL, NULL);
#elif defined(HAVE_SYSCONF) && defined(_SC_NPROCESSORS_ONLN) #elif defined(HAVE_SYSCONF) && defined(_SC_NPROCESSORS_ONLN)