foxhunt/tests/gpu/mod.rs

//! GPU Testing Infrastructure for Foxhunt HFT System
//! 
//! This module provides comprehensive GPU testing coverage including:
//! - CUDA device initialization and detection
//! - GPU memory management validation
//! - ML model GPU inference testing
//! - Performance benchmarks (GPU vs CPU)
//! - CUDA kernel validation
//! - Production GPU path testing

pub mod cuda_initialization_test;
pub mod gpu_memory_management_test;
pub mod ml_gpu_inference_test;
pub mod cuda_kernel_test;
pub mod gpu_performance_bench;
pub mod production_gpu_integration_test;

// Re-export commonly used test utilities
pub use cuda_initialization_test::*;
pub use gpu_memory_management_test::*;
pub use ml_gpu_inference_test::*;
pub use cuda_kernel_test::*;
pub use gpu_performance_bench::*;
pub use production_gpu_integration_test::*;

/// Common GPU test utilities
pub mod utils {
    use std::sync::Once;
    
    static INIT: Once = Once::new();
    
    /// Initialize GPU test environment once per test run
    pub fn init_gpu_test_env() {
        INIT.call_once(|| {
            env_logger::init();
            log::info!("🚀 Initializing GPU test environment");
        });
    }
    
    /// Check if CUDA is available for testing
    pub fn cuda_available() -> bool {
        #[cfg(feature = "cuda")]
        {
            cudarc::driver::CudaDevice::new(0).is_ok()
        }
        #[cfg(not(feature = "cuda"))]
        {
            false
        }
    }
    
    /// Get test device (CUDA if available, CPU otherwise)
    pub fn get_test_device() -> candle_core::Device {
        candle_core::Device::cuda_if_available(0).unwrap_or(candle_core::Device::Cpu)
    }
    
    /// Skip test if CUDA not available
    pub fn require_cuda() {
        if !cuda_available() {
            eprintln!("⚠️  Skipping CUDA test - CUDA not available");
            std::process::exit(0);
        }
    }
}