Struct InputSpecializationRequirementsBuilder 

pub struct InputSpecializationRequirementsBuilder { /* private fields */ }

Builder for per-input specialization requirements.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.dtype(true);
assert!(builder.build()?.specializes_dtype());

Implementations

impl InputSpecializationRequirementsBuilder

pub fn new() -> Self

Return a builder with no specialization fields enabled.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

assert!(!InputSpecializationRequirementsBuilder::new()
    .build()
    ?
    .specializes_dtype());

pub fn dtype(&mut self, dtype: bool) -> &mut Self

Set whether dtype should be specialized.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.dtype(true);
assert!(builder.build()?.specializes_dtype());

pub fn rank(&mut self, rank: bool) -> &mut Self

Set whether rank should be specialized.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.rank(true);
assert!(builder.build()?.specializes_rank());

pub fn concrete_dimensions(&mut self, axes: impl Into<Vec<u32>>) -> &mut Self

Set concrete axes whose dimensions should be specialized.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.rank(true).concrete_dimensions(vec![0]);
assert_eq!(builder.build()?.concrete_dimensions(), &[0]);

pub fn placement(&mut self, placement: PlacementSpecialization) -> &mut Self

Set placement specialization.

Examples

use tenferro_runtime::{
    InputSpecializationRequirementsBuilder, PlacementSpecialization,
};

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.placement(PlacementSpecialization::StorageClass);
assert_eq!(builder.build()?.placement(), PlacementSpecialization::StorageClass);

pub fn layout(&mut self, layout: LayoutSpecialization) -> &mut Self

Set layout specialization.

Examples

use tenferro_runtime::{InputSpecializationRequirementsBuilder, LayoutSpecialization};

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.rank(true).layout(LayoutSpecialization::ExactStrides);
assert_eq!(builder.build()?.layout(), LayoutSpecialization::ExactStrides);

pub fn alignment_log2(&mut self, alignment_log2: Option<u8>) -> &mut Self

Set a required alignment class.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let mut builder = InputSpecializationRequirementsBuilder::new();
builder.alignment_log2(Some(1));
assert_eq!(builder.build()?.alignment_log2(), Some(1));

pub fn build( self, ) -> Result<InputSpecializationRequirements, InputSpecializationRequirementsError>

Validate and build per-input requirements.

Examples

use tenferro_runtime::InputSpecializationRequirementsBuilder;

let requirements = InputSpecializationRequirementsBuilder::new().build()?;
assert!(requirements.concrete_dimensions().is_empty());

Errors

Returns InputSpecializationRequirementsError when the requested fields are internally inconsistent or outside finite lattices.

Trait Implementations

impl Clone for InputSpecializationRequirementsBuilder

fn clone(&self) -> InputSpecializationRequirementsBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InputSpecializationRequirementsBuilder

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

Formats the value using the given formatter.

impl Default for InputSpecializationRequirementsBuilder

fn default() -> InputSpecializationRequirementsBuilder

Returns the “default value” for a type.

impl PartialEq for InputSpecializationRequirementsBuilder

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for InputSpecializationRequirementsBuilder

impl StructuralPartialEq for InputSpecializationRequirementsBuilder