Struct cubecl_core::tune::TuneBenchmark

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

A benchmark that runs on server handles

Implementations

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

where S: ComputeServer,

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

Constructs a new TuneBenchmark.

Trait Implementations

impl<S, C, Out> Benchmark for TuneBenchmark<S, C, Out>

where S: ComputeServer, C: ComputeChannel<S>,

type Args = Box<dyn AutotuneOperation<Out>>

Benchmark arguments.

fn prepare(&self) -> <TuneBenchmark<S, C, Out> as Benchmark>::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: <TuneBenchmark<S, C, Out> as Benchmark>::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.