Question

我试图弄清楚如何使用win32 API设置std :: thread或boost :: thread的线程亲和力。我想使用SetThreadAffinityMask函数将每个线程固定到计算机中的特定内核。

我使用了线程native_handle成员函数来获取提供给SetThreadAffinityMask函数的线程句柄。但是，执行此操作将导致SetThreadAffinityMask函数返回0，表示无法设置线程相似性。

unsigned numCores = std::thread::hardware_concurrency();
std::vector<std::thread> threads(numCores);

for (int i = 0; i < numCores; i++)
{
    threads.push_back(std::thread(workLoad, i));
    cout << "Original Thread Affinity Mask: " << SetThreadAffinityMask(threads[i].native_handle() , 1 << i) << endl;

}

for (thread& t : threads)
{
    if (t.joinable())
        t.join();
}

原始线程关联性掩码：0

...等

Answer 1

您的问题是threads的初始设置，其中包含numCores默认初始化的条目。新线程（已读：实数）随后被推到向量上，但是在设置亲和力时，您永远不会索引到它们。取而代之的是，您使用i进行索引，该索引会在向量中击中实际运行线程之前真正运行的线程中的对象。

实际上值得运行的更正版本显示如下：

#include <iostream>
#include <vector>
#include <thread>
#include <chrono>

#include <windows.h>

void proc(void)
{
    using namespace std::chrono_literals;
    std::this_thread::sleep_for(5s);
}

int main()
{
    std::vector<std::thread> threads;
    for (unsigned int i = 0; i < std::thread::hardware_concurrency(); ++i)
    {
        threads.emplace_back(proc);
        DWORD_PTR dw = SetThreadAffinityMask(threads.back().native_handle(), DWORD_PTR(1) << i);
        if (dw == 0)
        {
            DWORD dwErr = GetLastError();
            std::cerr << "SetThreadAffinityMask failed, GLE=" << dwErr << '\n';
        }
    }

    for (auto& t : threads)
        t.join();
}

设置std :: threads

1 个答案: