Struct Fused 

pub struct Fused(/* private fields */);

A graph of element-wise operations evaluated as one kernel, with one or more of its steps as results. Cheap to clone; equal graphs compare and hash equal.

Implementations

impl Fused

pub fn supports(primitive: &Primitive, dtypes: &[DType]) -> bool

Whether primitive applied to inputs of these dtypes can be part of a fused kernel: unary, binary, select and type conversions over the GPU’s element types (all but float64 and complex64).

pub fn new( inputs: &[DType], steps: Vec<(Primitive, Vec<FusedRef>)>, outputs: &[usize], ) -> Result<Self>

The fused computation of steps over inputs of the given dtypes, with the results of the steps at positions outputs as its outputs. Each step is a primitive Fused::supports and its operands, which refer to inputs or earlier steps.

pub fn with_constants( inputs: &[DType], constants: &[(DType, u64)], steps: Vec<(Primitive, Vec<FusedRef>)>, outputs: &[usize], ) -> Result<Self>

Fused::new with scalar constants (a float dtype, F32, F16 or BF16, and the value’s bits) that steps refer to as FusedRef::Constant: they are part of the kernel, as literals, rather than inputs read from memory.

pub fn inputs(&self) -> &[DType]

The input dtypes.

pub fn len(&self) -> usize

The number of fused operations.

pub fn is_empty(&self) -> bool

Always false: a fused computation has at least one step.

pub fn dtypes(&self) -> Vec<DType>

The outputs’ dtypes.

Trait Implementations

impl Clone for Fused

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Fused

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

Formats the value using the given formatter.

impl Eq for Fused

impl Hash for Fused

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for Fused

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for Fused