threadpool_wrapper.hpp

/*
 * SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <array>
#include <atomic>
#include <cstdio>
#include <functional>
#include <memory>
#include <string>
 
#include <BS_thread_pool.hpp>
 
#include <pthread.h>
 
namespace kvikio {
 
using ThreadPool = BS::thread_pool;
 
[[nodiscard]] inline std::function<void()> make_thread_pool_init_task(std::string prefix)
{
  auto counter = std::make_shared<std::atomic<unsigned int>>(0);
  return [counter = std::move(counter), prefix = std::move(prefix)]() {
    unsigned int const idx = counter->fetch_add(1, std::memory_order_relaxed);
    // Linux comm limit is 15 chars + NUL.
    std::array<char, 16> name{};
    std::snprintf(name.data(), name.size(), "%s-%u", prefix.c_str(), idx);
    pthread_setname_np(pthread_self(), name.data());
  };
}
 
}  // namespace kvikio