Struct GPUArrayBuilder

pub struct GPUArrayBuilder { /* private fields */ }

A builder for GPU arrays

Implementations

impl GPUArrayBuilder

pub fn new() -> Self

Create a new GPU array builder with default settings.

pub const fn backend(self, backend: GPUBackend) -> Self

Set the GPU backend to use.

pub const fn device_id(self, device_id: usize) -> Self

Set the device ID to use.

pub const fn async_ops(self, async_ops: bool) -> Self

Set whether to use asynchronous operations.

pub const fn mixed_precision(self, mixed_precision: bool) -> Self

Set whether to use mixed precision.

pub const fn memory_fraction(self, memory_fraction: f32) -> Self

Set the fraction of GPU memory to use.

pub fn build<T, D>(self, host_data: Array<T, D>) -> GPUNdarray<T, D>

where T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T>, D: Dimension + Clone + Send + Sync + 'static + RemoveAxis,

Build a GPU array from a host array.

Trait Implementations

impl Default for GPUArrayBuilder

fn default() -> Self

Returns the “default value” for a type.