//! GPU Scheduler Module (PMAT-802)
//!
//! Shattered from gpu/cuda_scheduler.rs
//! Types extracted to types.rs for file health (PMAT-COMPLY)

pub mod batch;
mod core;
mod kv;
mod loading;
mod model;
mod ops;
mod types;

#[cfg(feature = "cuda")]
pub use core::CudaScheduler;
pub use model::GpuModel;
pub use types::{AttentionBuffers, BlockWeights, GpuGenerateConfig, GpuModelConfig, WeightType};

// Re-export KV cache functions for external use (public API)
#[allow(unused_imports)]
pub use kv::{forward_gpu_incremental, forward_gpu_with_cache, generate_with_cache};
