forEachExample.cpp

Go to the documentation of this file.

/**
 * @file
 * @copyright
 *   SPDX-FileCopyrightText: 2023-2023 European Southern Observatory (ESO)
 *
 *   SPDX-License-Identifier: LGPL-3.0-only
 */
#include <chrono>
#include <iostream>
 
#include <numapp/cpumask.hpp>
#include <numapp/numa.hpp>
#include <numapp/thread.hpp>
 
void ThreadFunc(std::chrono::milliseconds duration) {
    std::cout << "Thread affinity: " << numapp::CpuAffinity::MakeFromActive() << std::endl;
 
    std::this_thread::sleep_for(duration);
}
 
int main() {
    using namespace numapp;
    using namespace std::chrono_literals;
    if (!NumaAvailable()) {
        std::cout << "NUMA not available";
        return -1;
    }
 
    std::vector<std::thread> threads;
    std::string name = "thread-";
    // Launch thread for each CPU in CPU string
    ForEach(Cpumask::MakeFromCpuStringAll("0-3"), [&](unsigned int cpu) {
        threads.push_back(MakeThread(
            name + std::to_string(cpu), CpuAffinity::MakeBindCpu(cpu), &ThreadFunc, 500ms));
    });
 
    for (auto& thread : threads) {
        thread.join();
    }
}