Struct Device

pub struct Device { /* private fields */ }

Representation of a Device in MLX.

Implementations

impl Device

pub fn new(device_type: DeviceType, index: i32) -> Device

Create a new Device

pub fn try_default() -> Result<Self>

Try to get the default device.

pub fn cpu() -> Device

Create a default CPU device.

pub fn gpu() -> Device

Create a default GPU device.

pub fn set_default(device: &Device)

Set the default device.

Example:

use mlx_rs::{Device, DeviceType};
Device::set_default(&Device::new(DeviceType::Cpu, 1));

By default, this is gpu().

Trait Implementations

impl Debug for Device

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

Formats the value using the given formatter.

impl Default for Device

fn default() -> Self

Returns the “default value” for a type.

impl Display for Device

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

Formats the value using the given formatter.

impl Drop for Device

fn drop(&mut self)

Executes the destructor for this type.