Struct AlgorithmSelector 

pub struct AlgorithmSelector { /* private fields */ }

The main entry point for cuBLASLt-style algorithm selection.

An AlgorithmSelector is constructed for a specific GPU architecture and pre-populates a catalogue of known GEMM algorithm variants. Users can then enumerate compatible algorithms, query heuristics, or let the selector auto-pick the best candidate.

Implementations

impl AlgorithmSelector

pub fn new(sm_version: SmVersion) -> Self

Creates a new selector targeting the given SM architecture.

The selector is pre-populated with a set of built-in algorithm variants covering SIMT, Tensor Core, skinny, split-K, stream-K, warp-specialized, and bandwidth-limited categories.

pub fn enumerate_algorithms(&self, problem: &GemmProblem) -> Vec<AlgorithmId>

Enumerates all algorithms compatible with the given problem.

An algorithm is compatible if:

• The problem dimensions are non-zero.
• The problem’s math mode allows the algorithm’s compute path.
• Architecture-specific requirements are met (e.g., Tensor Core needs MathMode::TensorCore and an eligible type).

pub fn get_heuristic( &self, algo_id: &AlgorithmId, problem: &GemmProblem, ) -> BlasResult<AlgorithmHeuristic>

Returns a performance heuristic for a specific algorithm on the given problem.

Errors

Returns BlasError::InvalidArgument if the algorithm ID is not found in the selector’s catalogue.

pub fn select_best(&self, problem: &GemmProblem) -> BlasResult<AlgorithmId>

Automatically selects the best algorithm for the given problem.

Evaluates heuristics for all compatible algorithms and returns the one with the highest composite score.

Errors

Returns BlasError::UnsupportedOperation if no compatible algorithm is found.

pub fn select_top_n( &self, problem: &GemmProblem, n: usize, ) -> Vec<(AlgorithmId, AlgorithmHeuristic)>

Returns the top N algorithms for the given problem, ranked by heuristic score (descending).

If fewer than n algorithms are compatible, returns all of them.

pub fn create_custom( &mut self, config: AlgorithmConfig, problem: &GemmProblem, ) -> BlasResult<AlgorithmId>

Creates a custom algorithm with user-specified configuration.

The configuration is validated against the problem dimensions.

Errors

Returns BlasError::InvalidArgument if the configuration is invalid.

pub fn algorithm_count(&self) -> usize

Returns the number of registered algorithms.

pub fn sm_version(&self) -> SmVersion

Returns the target SM version.