Struct DeviceHandle 

pub struct DeviceHandle<P: PhantomDevice> { /* private fields */ }

Type-safe device handle with phantom device information

This wrapper provides compile-time device type safety for device operations. The phantom type parameter carries device information at the type level.

Examples

use torsh_core::device::{DeviceHandle, PhantomCpu, PhantomCuda};

// CPU device handle
let cpu_handle = DeviceHandle::<PhantomCpu>::new(cpu_device);

// CUDA device handle with index 0
let cuda_handle = DeviceHandle::<PhantomCuda<0>>::new(cuda_device);

// Compile-time device type checking
assert!(cpu_handle.is_cpu());
assert!(cuda_handle.is_gpu());

Implementations

impl<P: PhantomDevice> DeviceHandle<P>

pub fn new(device: Box<dyn Device>) -> Result<Self, TorshError>

Create a new device handle with phantom type information

pub unsafe fn new_unchecked(device: Box<dyn Device>) -> Self

Create an unchecked device handle (unsafe)

Safety

The caller must ensure that the device type matches the phantom type.

pub fn device(&self) -> &dyn Device

Get the underlying device

pub fn device_mut(&mut self) -> &mut dyn Device

Get the underlying device mutably

pub fn phantom_device_type() -> DeviceType

Get the phantom device type

pub const fn is_cpu() -> bool

Check if this is a CPU device (compile-time)

pub const fn is_gpu() -> bool

Check if this is a GPU device (compile-time)

pub const fn is_cuda() -> bool

Check if this is a CUDA device (compile-time)

pub const fn is_metal() -> bool

Check if this is a Metal device (compile-time)

pub const fn is_wgpu() -> bool

Check if this is a WebGPU device (compile-time)

pub fn cast<Q: PhantomDevice>( self, ) -> Result<DeviceHandle<Q>, (Self, TorshError)>

Convert to a different phantom device type (with runtime check)

pub unsafe fn cast_unchecked<Q: PhantomDevice>(self) -> DeviceHandle<Q>

Convert to a different phantom device type (unsafe, no runtime check)

Safety

The caller must ensure that the device type matches the target phantom type.

pub fn into_device(self) -> Box<dyn Device>

Extract the underlying device, consuming the handle

Trait Implementations

impl<P: PhantomDevice> Clone for DeviceHandle<P>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<P: Debug + PhantomDevice> Debug for DeviceHandle<P>

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

Formats the value using the given formatter.