Struct ScoreFunction 

pub struct ScoreFunction {
    pub tc_weight: f64,
    pub sc_weight: f64,
    pub rw_weight: f64,
    pub sc_target: f64,
}

Weighted scoring function for contraction quality evaluation.

The score is computed as:

score = tc_weight * 2^tc + rw_weight * 2^rw + sc_weight * max(0, 2^sc - 2^sc_target)

Where:

• tc is the time complexity (log2 of FLOP count)
• sc is the space complexity (log2 of max intermediate tensor size)
• rw is the read-write complexity (log2 of total I/O operations)

Example

use omeco::ScoreFunction;

let score = ScoreFunction::default();
let result = score.evaluate(10.0, 5.0, 8.0);

Fields

tc_weight: f64

Weight for time complexity (default: 1.0)

sc_weight: f64

Weight for space complexity penalty (default: 1.0)

rw_weight: f64

Weight for read-write complexity (default: 0.0)

sc_target: f64

Target space complexity threshold (default: 20.0) Space complexity is only penalized if it exceeds this target.

Implementations

impl ScoreFunction

pub fn new( tc_weight: f64, sc_weight: f64, rw_weight: f64, sc_target: f64, ) -> Self

Create a new ScoreFunction with custom weights.

pub fn time_optimized() -> Self

Create a score function optimizing primarily for time complexity.

pub fn space_optimized(sc_target: f64) -> Self

Create a score function optimizing primarily for space complexity.

pub fn evaluate(&self, tc: f64, sc: f64, rw: f64) -> f64

Evaluate the score for given complexity values.

Arguments

• tc - Time complexity (log2)
• sc - Space complexity (log2)
• rw - Read-write complexity (log2)

Returns

The weighted score value (lower is better).

pub fn exceeds_target(&self, sc: f64) -> bool

Check if space complexity exceeds the target.

pub fn with_sc_target(self, sc_target: f64) -> Self

Builder method to set the space complexity target.

pub fn with_tc_weight(self, tc_weight: f64) -> Self

Builder method to set the time complexity weight.

pub fn with_sc_weight(self, sc_weight: f64) -> Self

Builder method to set the space complexity weight.

pub fn with_rw_weight(self, rw_weight: f64) -> Self

Builder method to set the read-write complexity weight.

Trait Implementations

impl Clone for ScoreFunction

fn clone(&self) -> ScoreFunction

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ScoreFunction

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

Formats the value using the given formatter.

impl Default for ScoreFunction

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ScoreFunction

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ScoreFunction

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.