Struct CudaPlan 

pub struct CudaPlan { /* private fields */ }

A graph lowered for one bucket on one GPU.

Implementations

impl CudaPlan

pub fn bucket(&self) -> Bucket

The bucket it was built for.

pub fn arena_bytes(&self) -> usize

Arena size in bytes.

pub fn profile(&mut self)

Times every step from now on. Each step then waits for the GPU and the graph is not used, so this is for finding where the time goes, not for measuring the total.

pub fn gemm_timings(&self) -> Vec<((usize, usize, usize), u64, u64)>

Time per GEMM shape since CudaPlan::profile: rows, inner size, columns, calls over all timed runs and nanoseconds, largest first.

pub fn timings(&self) -> Vec<(&'static str, u64)>

Time per kind of step since CudaPlan::profile, in nanoseconds, largest first.

Trait Implementations

impl Debug for CudaPlan

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

Formats the value using the given formatter.