Struct cubecl_runtime::tune::TuneBenchmark

pub struct TuneBenchmark<S: ComputeServer, C, Out = ()> { /* private fields */ }

A benchmark that runs on server handles

Implementations

impl<S: ComputeServer, C, Out> TuneBenchmark<S, C, Out>

pub fn new( operation: Box<dyn AutotuneOperation<Out>>, client: ComputeClient<S, C>, ) -> Self

Constructs a new TuneBenchmark.

Trait Implementations

impl<S: ComputeServer, C: ComputeChannel<S>, Out> Benchmark for TuneBenchmark<S, C, Out>

type Args = Box<dyn AutotuneOperation<Out>>

Benchmark arguments.

fn prepare(&self) -> Self::Args

Prepare the benchmark, run anything that is essential for the benchmark, but shouldn’t count as included in the duration.

fn num_samples(&self) -> usize

Number of samples per run required to have a statistical significance.

fn execute(&self, operation: Self::Args)

Execute the benchmark and returns the time it took to complete.

fn name(&self) -> String

Name of the benchmark, should be short and it should match the name defined in the crate Cargo.toml

fn sync(&self)

Wait for computed to be over

fn options(&self) -> Option<String>

The options passed to the benchmark.

fn shapes(&self) -> Vec<Vec<usize>>

Shapes dimensions

fn run(&self) -> BenchmarkDurations

Run the benchmark a number of times.