Trait HybridWorkload

Source

pub trait HybridWorkload: Send + Sync {
    type Result;

    // Required methods
    fn workload_size(&self) -> usize;
    fn execute_cpu(&self) -> Self::Result;
    fn execute_gpu(&self) -> HybridResult<Self::Result>;

    // Provided methods
    fn name(&self) -> &str { ... }
    fn supports_gpu(&self) -> bool { ... }
    fn memory_estimate(&self) -> usize { ... }
}

Expand description

Trait for workloads that can be executed on CPU or GPU.

Implementors provide both CPU and GPU execution paths, allowing the HybridDispatcher to choose the optimal backend based on workload size and runtime measurements.

§Example

use ringkernel_core::hybrid::{HybridWorkload, HybridResult};

struct VectorAdd {
    a: Vec<f32>,
    b: Vec<f32>,
}

impl HybridWorkload for VectorAdd {
    type Result = Vec<f32>;

    fn workload_size(&self) -> usize {
        self.a.len()
    }

    fn execute_cpu(&self) -> Self::Result {
        self.a.iter().zip(&self.b).map(|(a, b)| a + b).collect()
    }

    fn execute_gpu(&self) -> HybridResult<Self::Result> {
        // GPU implementation
        todo!("GPU kernel execution")
    }
}