[并发并行]_[Windows]_[指定程序执行的CPU]

场景

1. 在编写音视频转换或者需要CPU密集型运算的应用时,一般会开启多线程进行并发执行,而多线程往往就会使用多个CPU. 这时候如果把CPU都用完了,CPU 使用率达到很高的话,就会影响电脑桌面的其他任务的执行,比如点击界面的软件,鼠标移动都会卡.

2. 这时候如果只使用一半的CPU的话应该就不会影响用户的正常操作了.

3. Win32 函数 SetProcessAffinityMask 就可以解决这个问题.

参考

用SetProcessAffinityMask为进程指定CPU
SetProcessAffinityMask function
Example usage of SetProcessAffinityMask in C/C++

例子

这个例子测试只使用第3,4的内核执行当前进程任务.

// test_cpu.cpp : 定义控制台应用程序的入口点。//#include "stdafx.h"#include <Windows.h>#include <time.h>#include <assert.h>#include <iostream>bool SetWorkProcessor(int n,...){    // On a system with more than 64 processors,    // the affinity mask must specify processors in a single processor group    assert(n < 64);    HANDLE process = GetCurrentProcess();    DWORD_PTR processAffinityMask;    DWORD_PTR systemAffinityMask;    if (!GetProcessAffinityMask(process, &processAffinityMask, &systemAffinityMask)) return false;    // 获取CPU个数    SYSTEM_INFO info;    ::GetSystemInfo(&info);    // 设置处理器    va_list argptr;    va_start(argptr, n);    int result = 0;    for(int i = 0;i<n;++i)    {        int value = va_arg(argptr, int);        if(value < info.dwNumberOfProcessors)        {            result |= (1 << value);        }    }    va_end(argptr);    if(result)    {        return SetProcessAffinityMask(process,result);    }    return false;}DWORD WINAPI ThreadFunc(PVOID pvParam){    int i = 0;    while(true)    {        ++i;    }    return 0;}int _tmain(int argc, _TCHAR* argv[]){    // 设置使用的CPU    assert(SetWorkProcessor(2,2,3));    DWORD dwThreadID;    for(int i = 0; i < 10; ++i)    {        CreateThread(NULL,0,ThreadFunc,NULL,0,&dwThreadID);    }    int i = 0;    while(true)    {        ++i;    }    return 0;}

图1 未执行程序前,3,4 CPU使用率并不高

图2 执行程序后,3,4 CPU使用率暴涨,但是1,2 CPU没怎么变化.

图3 不设置CPU处理函数,看4个CPU使用率吓人.