Device

Trait Device 

Source
pub trait Device:
    Default
    + Clone
    + Debug
    + Send
    + Sync
    + 'static {
    // Required methods
    fn from_id(device_id: DeviceId) -> Self;
    fn to_id(&self) -> DeviceId;
    fn device_count(type_id: u16) -> usize;

    // Provided method
    fn device_count_total() -> usize { ... }
}
Expand description

Device trait for all cubecl devices.

Required Methods§

Source

fn from_id(device_id: DeviceId) -> Self

Create a device from its id.

Source

fn to_id(&self) -> DeviceId

Retrieve the device id from the device.

Source

fn device_count(type_id: u16) -> usize

Returns the number of devices available under the provided type id.

Provided Methods§

Source

fn device_count_total() -> usize

Returns the total number of devices that can be handled by the runtime.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§