From fd2221e50ec3b8bb27f808eda61b248bdaef0aba Mon Sep 17 00:00:00 2001
From: jgrusewski <jgrusewski@localhost>
Date: Mon, 23 Feb 2026 11:41:10 +0100
Subject: [PATCH] feat(ml): export GPU types in prelude

Add DeviceConfig and GpuCapabilities re-exports to ml::prelude so
downstream crates can import GPU management types via a single
`use ml::prelude::*` statement.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 ml/src/lib.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ml/src/lib.rs b/ml/src/lib.rs
index bcf2f460d..26ff71d7f 100644
--- a/ml/src/lib.rs
+++ b/ml/src/lib.rs
@@ -1950,6 +1950,9 @@ pub mod prelude {
     // Constants
     pub use crate::{MAX_INFERENCE_LATENCY_US, PRECISION_FACTOR};
 
+    // GPU device management
+    pub use crate::gpu::{capabilities::GpuCapabilities, DeviceConfig};
+
     // Tensor types from candle
     pub use candle_core::{Device, Tensor};
     pub use candle_nn::{Module, VarBuilder, VarMap};