Struct TunerFeaturesBuilder 

pub struct TunerFeaturesBuilder { /* private fields */ }

Builder for TunerFeatures with automatic normalization.

Implementations

impl TunerFeaturesBuilder

pub fn model_params_b(self, params: f32) -> Self

Set model size in billions of parameters

pub fn hidden_dim(self, dim: u32) -> Self

Set hidden dimension

pub fn num_layers(self, layers: u32) -> Self

Set number of layers

pub fn num_heads(self, heads: u32) -> Self

Set number of attention heads

pub fn head_dim(self, dim: u32) -> Self

Set head dimension

pub fn vocab_size(self, size: u32) -> Self

Set vocabulary size

pub fn batch_size(self, m: u32) -> Self

Set batch size (M)

pub fn seq_len(self, len: u32) -> Self

Set sequence length

pub fn cuda_graphs(self, enabled: bool) -> Self

Enable CUDA graphs

pub fn kv_caches(self, count: u32) -> Self

Set number of KV caches

pub fn is_prefill(self, prefill: bool) -> Self

Set prefill mode

pub fn quant_type(self, qt: QuantType) -> Self

Set quantization type

pub fn kernel_type(self, kt: KernelType) -> Self

Set kernel type

pub fn gpu_mem_bw_gbs(self, bw: f32) -> Self

Set GPU memory bandwidth in GB/s

pub fn gpu_compute_tflops(self, tflops: f32) -> Self

Set GPU compute in TFLOPS

pub fn gpu_sm_count(self, count: u32) -> Self

Set GPU SM count

pub fn measured_tps(self, tps: f32) -> Self

Set measured throughput (for training data)

pub fn gpu_l2_cache_mb(self, l2_mb: f32) -> Self

Set L2 cache size in MB (v1.1.0)

pub fn is_zero_copy(self, enabled: bool) -> Self

Set zero-copy memory path enabled (v1.1.0)

pub fn hardware(self, hw: &HardwareCapability) -> Self

Set hardware capability (auto-fills GPU features)

pub fn build(self) -> TunerFeatures

Build the feature vector with normalization Build features, panicking if raw inputs violate physical constraints.

Use try_build() for fallible construction.

pub fn try_build(self) -> Result<TunerFeatures, TunerError>

Trait Implementations

impl Default for TunerFeaturesBuilder

fn default() -> TunerFeaturesBuilder

Returns the “default value” for a type.