Struct DeviceArray 

pub struct DeviceArray<T> { /* private fields */ }

A minimal, shaped device array with dtype. Currently CPU-backed.

Implementations

impl<T: Copy> DeviceArray<T>

pub fn from_cpu_slice(shape: &[usize], dtype: DType, data: &[T]) -> Self

Create a DeviceArray from a CPU slice and explicit shape/dtype.

Examples

let data = vec![1.0f32, 2.0, 3.0, 4.0];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[2,2], scir_gpu::DType::F32, &data);
assert_eq!(arr.shape(), &[2,2]);

pub fn to_cpu_vec(&self) -> Vec<T>

Copy data back to a CPU-owned Vec<T>.

Examples

let data = vec![1i32,2,3];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &data);
let back = arr.to_cpu_vec();
assert_eq!(back, vec![1,2,3]);

pub fn shape(&self) -> &[usize]

Return the logical shape of the array.

Examples

let data = vec![0u8; 6];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[2,3], scir_gpu::DType::F32, &data);
assert_eq!(arr.shape(), &[2,3]);

pub fn dtype(&self) -> DType

Return the element data type.

Examples

let data = vec![0u8; 4];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[4], scir_gpu::DType::F32, &data);
assert!(matches!(arr.dtype(), scir_gpu::DType::F32));

pub fn device(&self) -> Device

Return the current device of this array.

Examples

let data = vec![1u8,2,3,4];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[4], scir_gpu::DType::F32, &data);
assert!(matches!(arr.device(), scir_gpu::Device::Cpu));

impl<T: Copy> DeviceArray<T>

pub fn to_device(&mut self, device: Device) -> Result<(), GpuError>

Move the array to a device (CPU only, when CUDA is disabled).

impl<T> DeviceArray<T>

where T: Copy + NumAssign,

pub fn add_scalar(&self, alpha: T) -> Self

Add a scalar to each element (CPU baseline).

Examples

let data = vec![1.0f32, 2.0, 3.0];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &data);
let out = arr.add_scalar(1.0f32);
assert_eq!(out.to_cpu_vec(), vec![2.0f32, 3.0, 4.0]);

pub fn mul_scalar(&self, alpha: T) -> Self

Multiply each element by a scalar (CPU baseline).

Examples

let data = vec![1.0f32, 2.0, 3.0];
let arr = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &data);
let out = arr.mul_scalar(2.0f32);
assert_eq!(out.to_cpu_vec(), vec![2.0f32, 4.0, 6.0]);

impl<T> DeviceArray<T>

where T: Copy + NumAssign,

pub fn add(&self, other: &Self) -> Result<Self, GpuError>

Elementwise addition between arrays (CPU baseline).

Examples

let a = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &[1.0f32,2.0,3.0]);
let b = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &[0.5f32,1.5,2.5]);
let c = a.add(&b).unwrap();
assert_eq!(c.to_cpu_vec(), vec![1.5f32, 3.5, 5.5]);

impl DeviceArray<f32>

pub fn add_scalar_auto(&self, alpha: f32) -> Self

Elementwise add-scalar, with CUDA dispatch when available.

Examples

let data = vec![1.0f32, 2.0, 3.0];
let mut a = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &data);
a.to_device(scir_gpu::Device::Cpu).unwrap();
let out = a.add_scalar_auto(1.0);
assert_eq!(out.to_cpu_vec(), vec![2.0f32, 3.0, 4.0]);

pub fn add_auto(&self, other: &Self) -> Result<Self, GpuError>

Elementwise add of two arrays, with CUDA dispatch when available.

Examples

let a = scir_gpu::DeviceArray::from_cpu_slice(&[2], scir_gpu::DType::F32, &[1.0f32,2.0]);
let b = scir_gpu::DeviceArray::from_cpu_slice(&[2], scir_gpu::DType::F32, &[0.5f32,1.5]);
let c = a.add_auto(&b).unwrap();
assert_eq!(c.to_cpu_vec(), vec![1.5f32, 3.5]);

pub fn mul_scalar_auto(&self, alpha: f32) -> Self

Elementwise mul-scalar with device dispatch.

Examples

let data = vec![1.0f32, 2.0, 3.0];
let a = scir_gpu::DeviceArray::from_cpu_slice(&[3], scir_gpu::DType::F32, &data);
let out = a.mul_scalar_auto(2.0);
assert_eq!(out.to_cpu_vec(), vec![2.0f32, 4.0, 6.0]);

Trait Implementations

impl<T: Clone> Clone for DeviceArray<T>

fn clone(&self) -> DeviceArray<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug> Debug for DeviceArray<T>

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

Formats the value using the given formatter.