Настройка схожести потоков в std :: threads

Question

Я пытаюсь выяснить, как установить сходство потоков с std :: thread или boost :: thread, используя win32 API. Я хочу использовать функцию SetThreadAffinityMask для закрепления каждого потока на определенном ядре в моей машине.

Я использовал функцию-член thread_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

Оригинальная маска сродства темы: 0

Оригинальная маска сродства темы: 0

Оригинальная маска сродства темы: 0

Оригинальная маска сродства темы: 0

Оригинальная маска сродства темы: 0

Оригинальная маска сродства темы: 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();
}