Struct PyDevice 

pub struct PyDevice { /* private fields */ }

Python wrapper for ToRSh devices

Represents a computational device where tensors can be allocated and operations executed. Supports CPU, CUDA (NVIDIA GPUs), Metal (Apple Silicon), and WGPU devices.

Examples

# Create CPU device
cpu = rstorch.PyDevice("cpu")

# Create CUDA device (default index 0)
cuda = rstorch.PyDevice("cuda")

# Create CUDA device with specific index
cuda1 = rstorch.PyDevice("cuda:1")

# Create from integer (defaults to CUDA)
cuda2 = rstorch.PyDevice(2)  # cuda:2

# Check device properties
print(cpu.type)     # "cpu"
print(cuda1.type)   # "cuda"
print(cuda1.index)  # 1

Trait Implementations

impl Clone for PyDevice

fn clone(&self) -> PyDevice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PyDevice

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

Formats the value using the given formatter.

impl Display for PyDevice

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

Formats the value using the given formatter.

impl From<DeviceType> for PyDevice

fn from(device: DeviceType) -> Self

Converts to this type from the input type.

impl From<PyDevice> for DeviceType

fn from(py_device: PyDevice) -> Self

Converts to this type from the input type.

impl<'a, 'py> FromPyObject<'a, 'py> for PyDevice

where Self: Clone,

type Error = PyClassGuardError<'a, 'py>

The type returned in the event of a conversion error.

fn extract( obj: Borrowed<'a, 'py, PyAny>, ) -> Result<Self, <Self as FromPyObject<'a, 'py>>::Error>

Extracts Self from the bound smart pointer obj.

impl<'py> IntoPyObject<'py> for PyDevice

type Target = PyDevice

The Python output type

type Output = Bound<'py, <PyDevice as IntoPyObject<'py>>::Target>

The smart pointer type to use.

type Error = PyErr

The type returned in the event of a conversion error.

fn into_pyobject( self, py: Python<'py>, ) -> Result<<Self as IntoPyObject<'_>>::Output, <Self as IntoPyObject<'_>>::Error>

Performs the conversion.

impl PyClass for PyDevice

const NAME: &str = "device"

Name of the class.

type Frozen = False

Whether the pyclass is frozen.

impl PyClassImpl for PyDevice

const MODULE: Option<&str> = ::core::option::Option::None

Module which the class will be associated with.

const IS_BASETYPE: bool = false

#[pyclass(subclass)]

const IS_SUBCLASS: bool = false

#[pyclass(extends=…)]

const IS_MAPPING: bool = false

#[pyclass(mapping)]

const IS_SEQUENCE: bool = false

#[pyclass(sequence)]

const IS_IMMUTABLE_TYPE: bool = false

#[pyclass(immutable_type)]

const RAW_DOC: &'static CStr = /// Python wrapper for ToRSh devices /// /// Represents a computational device where tensors can be allocated and operations executed. /// Supports CPU, CUDA (NVIDIA GPUs), Metal (Apple Silicon), and WGPU devices. /// /// # Examples /// /// ```python /// # Create CPU device /// cpu = rstorch.PyDevice("cpu") /// /// # Create CUDA device (default index 0) /// cuda = rstorch.PyDevice("cuda") /// /// # Create CUDA device with specific index /// cuda1 = rstorch.PyDevice("cuda:1") /// /// # Create from integer (defaults to CUDA) /// cuda2 = rstorch.PyDevice(2) # cuda:2 /// /// # Check device properties /// print(cpu.type) # "cpu" /// print(cuda1.type) # "cuda" /// print(cuda1.index) # 1 /// ```

Docstring for the class provided on the struct or enum.

const DOC: &'static CStr

Fully rendered class doc, including the text_signature if a constructor is defined.

type Layout = <<PyDevice as PyClassImpl>::BaseNativeType as PyClassBaseType>::Layout<PyDevice>

Description of how this class is laid out in memory

type BaseType = PyAny

Base class

type ThreadChecker = NoopThreadChecker

This handles following two situations:

type PyClassMutability = <<PyAny as PyClassBaseType>::PyClassMutability as PyClassMutability>::MutableChild

Immutable or mutable

type Dict = PyClassDummySlot

Specify this class has #[pyclass(dict)] or not.

type WeakRef = PyClassDummySlot

Specify this class has #[pyclass(weakref)] or not.

type BaseNativeType = PyAny

The closest native ancestor. This is PyAny by default, and when you declare #[pyclass(extends=PyDict)], it’s PyDict.

fn items_iter() -> PyClassItemsIter

fn lazy_type_object() -> &'static LazyTypeObject<Self>

fn dict_offset() -> Option<PyObjectOffset>

Used to provide the dictoffset slot (equivalent to tp_dictoffset)

fn weaklist_offset() -> Option<PyObjectOffset>

Used to provide the weaklistoffset slot (equivalent to tp_weaklistoffset

impl PyClassNewTextSignature for PyDevice

const TEXT_SIGNATURE: &'static str = "(device)"

impl PyClass__eq__SlotFragment<PyDevice> for PyClassImplCollector<PyDevice>

unsafe fn __eq__( self, py: Python<'_>, _slf: *mut PyObject, arg0: *mut PyObject, ) -> PyResult<*mut PyObject>

Safety: _slf and _other must be valid non-null Python objects

impl PyMethods<PyDevice> for PyClassImplCollector<PyDevice>

fn py_methods(self) -> &'static PyClassItems

impl PyTypeInfo for PyDevice

const NAME: &str = <Self as ::pyo3::PyClass>::NAME

👎Deprecated since 0.28.0:

prefer using ::type_object(py).name() to get the correct runtime value

Class name.

const MODULE: Option<&str> = <Self as ::pyo3::impl_::pyclass::PyClassImpl>::MODULE

👎Deprecated since 0.28.0:

prefer using ::type_object(py).module() to get the correct runtime value

Module name, if any.

fn type_object_raw(py: Python<'_>) -> *mut PyTypeObject

Returns the PyTypeObject instance for this type.

fn type_object(py: Python<'_>) -> Bound<'_, PyType>

Returns the safe abstraction over the type object.

fn is_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type or a subclass of this type.

fn is_exact_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type.

impl DerefToPyAny for PyDevice