scalar_bf16_or_f32

metrale_model_layers::weight_map::model_a

Function scalar_bf16_or_f32 

Source 
pub fn scalar_bf16_or_f32(
    store: &WeightStore,
    name: &str,
    gpu: &dyn GpuBackend,
) -> Result<f32>
Expand description

2026-09-28: Read a one-element BF16 or FP32 tensor to the host as f32; any other dtype or element count is an error. Checkpoints ship FP8 KV scales in both widths (unsloth: BF16 [1]; ModelOpt: FP32 []), and an F16 export is already BF16 in the store.