Struct DeviceConfig 

pub struct DeviceConfig {
    pub device_type: DeviceType,
    pub device_id: usize,
    pub use_fp16: bool,
    pub use_tf32: bool,
}

Device configuration for GPU acceleration

Fields

device_type: DeviceType

Device type to use

device_id: usize

Device ID (for multi-GPU systems)

use_fp16: bool

Enable mixed precision (FP16)

use_tf32: bool

Enable TF32 for matmul (CUDA only)

Implementations

impl DeviceConfig

pub fn new() -> Self

Create a new device configuration

pub fn with_device_type(self, device_type: DeviceType) -> Self

Set device type

pub fn with_device_id(self, device_id: usize) -> Self

Set device ID

pub fn with_fp16(self, enabled: bool) -> Self

Enable FP16 precision

pub fn with_tf32(self, enabled: bool) -> Self

Enable TF32 precision (CUDA only)

pub fn create_device(&self) -> CoreResult<Device>

Create a candle Device from this configuration

Trait Implementations

impl Clone for DeviceConfig

fn clone(&self) -> DeviceConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DeviceConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for DeviceConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for DeviceConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for DeviceConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.