Struct Bf16Einsum 

pub struct Bf16Einsum { /* private fields */ }

A pairwise contraction in bfloat16.

The labels are the ones an ordinary einsum subscript string names, so "ik,kj->ij" is (&[0, 1], &[1, 2], &[0, 2]). The operands are widened to f32, the contraction accumulates there, and the result is rounded to bfloat16 once.

Examples

use tenferro_bf16_proof::einsum::Bf16Einsum;

let op = Bf16Einsum::new(&[0, 1], &[1, 2], &[0, 2]).expect("a contraction");
assert_eq!(op.labels(), (&[0, 1][..], &[1, 2][..], &[0, 2][..]));

Implementations

impl Bf16Einsum

pub fn new(lhs: &[u32], rhs: &[u32], out: &[u32]) -> Result<Self>

Build the pairwise contraction lhs,rhs->out.

Errors

Returns [tenferro_tensor::Error::InvalidArgument] when the pattern is not a pairwise contraction: an operand carries no label, a label repeats inside one operand, the operands share no contracted label, or an output label appears in no operand.

Examples

use tenferro_bf16_proof::einsum::Bf16Einsum;

assert!(Bf16Einsum::new(&[0, 1], &[1, 2], &[0, 2]).is_ok());
assert!(Bf16Einsum::new(&[], &[1, 2], &[0, 2]).is_err());

pub fn labels(&self) -> (&[u32], &[u32], &[u32])

The pattern’s three label lists.

Examples

use tenferro_bf16_proof::einsum::Bf16Einsum;

let op = Bf16Einsum::new(&[0, 1], &[1, 2], &[0, 2]).expect("a contraction");
assert_eq!(op.labels(), (&[0, 1][..], &[1, 2][..], &[0, 2][..]));

Trait Implementations

impl Clone for Bf16Einsum

fn clone(&self) -> Bf16Einsum

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Bf16Einsum

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

Formats the value using the given formatter.

impl Eq for Bf16Einsum

impl ExtensionOp for Bf16Einsum

fn family_id(&self) -> &'static str

Stable, process-independent family identifier.

fn payload_hash(&self, hasher: &mut dyn Hasher)

Hash the payload (everything except family_id).

fn payload_eq(&self, other: &dyn ExtensionOp) -> bool

Structural equality against another extension value.

fn clone_arc(&self) -> Arc<dyn ExtensionOp>

Deep-clone the payload behind an Arc.

fn as_any(&self) -> &dyn Any

Upcast this extension to &dyn Any for downcasting in payload_eq.

fn input_count(&self) -> usize

Number of primal inputs. MUST be constant for any given Arc<dyn ExtensionOp> value.

fn output_count(&self) -> usize

Number of outputs. MUST match the length of the vector returned by a successful Self::infer_output_meta call.

fn semantic_effects(&self) -> ExtensionEffectDeclaration<'_>

Declare observable semantic effects for this extension payload.

fn semantic_aliases(&self) -> ExtensionAliasDeclaration<'_>

Declare semantic output aliasing for this extension payload.

fn scalar_identity(&self) -> Option<&'static str>

Canonical identity of an externally defined scalar this operation carries.

fn infer_output_meta( &self, ctx: &mut ExtensionShapeContext<'_>, ) -> Result<Vec<(DType, Vec<SymDim>)>>

Infer output dtypes and shapes for each output slot.

fn lower_to_standard_ops( &self, _builder: &mut GraphBuilder<StdTensorOp>, _inputs: &[ValueRef<StdTensorOp>], _input_dtypes: &[DType], _input_shapes: &[&[SymDim]], ) -> Result<ExtensionStandardLowering, ExtensionLoweringError>

Try to expand this extension into standard tensor graph operations.

fn prune_outputs(&self, _live_outputs: &[bool]) -> Option<Arc<dyn ExtensionOp>>

Optionally return an equivalent op that produces only live outputs.

impl PartialEq for Bf16Einsum

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for Bf16Einsum