enhanced_work_stealing_config.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include "enhanced_steal_policy.h"
#include "steal_backoff_strategy.h"
 
#include <chrono>
#include <cstddef>
 
namespace kcenon::thread
{
 
struct enhanced_work_stealing_config
{
    // ========================================================================
    // Enable/Disable
    // ========================================================================
 
    bool enabled = false;
 
    // ========================================================================
    // Victim Selection Policy
    // ========================================================================
 
    enhanced_steal_policy policy = enhanced_steal_policy::adaptive;
 
    // ========================================================================
    // NUMA Configuration
    // ========================================================================
 
    bool numa_aware = false;
 
    double numa_penalty_factor = 2.0;
 
    bool prefer_same_node = true;
 
    // ========================================================================
    // Batch Stealing Configuration
    // ========================================================================
 
    std::size_t min_steal_batch = 1;
 
    std::size_t max_steal_batch = 4;
 
    bool adaptive_batch_size = true;
 
    // ========================================================================
    // Steal Attempts Configuration
    // ========================================================================
 
    std::size_t max_steal_attempts = 3;
 
    std::size_t max_consecutive_failures = 10;
 
    // ========================================================================
    // Backoff Configuration
    // ========================================================================
 
    steal_backoff_strategy backoff_strategy = steal_backoff_strategy::exponential;
 
    std::chrono::microseconds initial_backoff{50};
 
    std::chrono::microseconds max_backoff{1000};
 
    double backoff_multiplier = 2.0;
 
    // ========================================================================
    // Locality Tracking
    // ========================================================================
 
    bool track_locality = false;
 
    std::size_t locality_history_size = 16;
 
    // ========================================================================
    // Statistics Collection
    // ========================================================================
 
    bool collect_statistics = false;
 
    // ========================================================================
    // Factory Methods
    // ========================================================================
 
    static auto default_config() -> enhanced_work_stealing_config
    {
        return enhanced_work_stealing_config{};
    }
 
    static auto numa_optimized() -> enhanced_work_stealing_config
    {
        enhanced_work_stealing_config config;
        config.enabled = true;
        config.policy = enhanced_steal_policy::numa_aware;
        config.numa_aware = true;
        config.prefer_same_node = true;
        config.numa_penalty_factor = 2.0;
        config.collect_statistics = true;
        return config;
    }
 
    static auto locality_optimized() -> enhanced_work_stealing_config
    {
        enhanced_work_stealing_config config;
        config.enabled = true;
        config.policy = enhanced_steal_policy::locality_aware;
        config.track_locality = true;
        config.locality_history_size = 32;
        config.collect_statistics = true;
        return config;
    }
 
    static auto batch_optimized() -> enhanced_work_stealing_config
    {
        enhanced_work_stealing_config config;
        config.enabled = true;
        config.policy = enhanced_steal_policy::adaptive;
        config.min_steal_batch = 2;
        config.max_steal_batch = 8;
        config.adaptive_batch_size = true;
        return config;
    }
 
    static auto hierarchical_numa() -> enhanced_work_stealing_config
    {
        enhanced_work_stealing_config config;
        config.enabled = true;
        config.policy = enhanced_steal_policy::hierarchical;
        config.numa_aware = true;
        config.prefer_same_node = true;
        config.numa_penalty_factor = 3.0;
        config.track_locality = true;
        config.collect_statistics = true;
        return config;
    }
};
 
} // namespace kcenon::thread