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>
This commit is contained in:
jgrusewski
2026-02-23 11:41:10 +01:00
parent 6425eb150b
commit fd2221e50e

View File

@@ -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};