Enum DeviceType 

pub enum DeviceType {
    Cpu,
    Cuda,
}

Device types supported by Train Station

This enum represents the different types of devices where tensor operations can be performed. Currently supports CPU and CUDA GPU devices.

Variants

• Cpu - CPU device for general-purpose computation
• Cuda - CUDA GPU device for accelerated computation (feature-gated)

Examples

use train_station::{DeviceType, Device};

let cpu_type = DeviceType::Cpu;
let cpu_device = Device::from(cpu_type);
assert!(cpu_device.is_cpu());

#[cfg(feature = "cuda")]
{
    let cuda_type = DeviceType::Cuda;
    let cuda_device = Device::from(cuda_type);
    assert!(cuda_device.is_cuda());
}

Thread Safety

This type is thread-safe and can be shared between threads.

Variants

Cpu

CPU device for general-purpose computation

Cuda

CUDA GPU device for accelerated computation (feature-gated)

Trait Implementations

impl Clone for DeviceType

fn clone(&self) -> DeviceType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeviceType

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

Formats the value using the given formatter.

impl Display for DeviceType

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

Format the device type as a string

Returns

String representation of the device type:

• "cpu" for CPU devices
• "cuda" for CUDA devices

impl From<DeviceType> for Device

fn from(device_type: DeviceType) -> Self

Convert DeviceType to Device with index 0

Arguments

• device_type - The device type to convert

Returns

A Device with the specified type and index 0

Panics

Panics if device_type is DeviceType::Cuda and CUDA is not available or the feature is not enabled.

Examples

use train_station::{Device, DeviceType};

let cpu_type = DeviceType::Cpu;
let cpu_device = Device::from(cpu_type);
assert!(cpu_device.is_cpu());
assert_eq!(cpu_device.index(), 0);

impl FromFieldValue for DeviceType

fn from_field_value( value: FieldValue, field_name: &str, ) -> SerializationResult<Self>

Convert FieldValue to DeviceType for deserialization

Arguments

• value - FieldValue containing enum data
• field_name - Name of the field for error reporting

Returns

DeviceType enum value or error if invalid

impl Hash for DeviceType

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DeviceType

fn eq(&self, other: &DeviceType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl ToFieldValue for DeviceType

fn to_field_value(&self) -> FieldValue

Convert DeviceType to FieldValue for serialization

Returns

Enum FieldValue with variant name (proper enum serialization)

impl Copy for DeviceType

impl Eq for DeviceType

impl StructuralPartialEq for DeviceType