tap_bf16

spark_model::layers::ple::dump

Function tap_bf16 

Source 
pub fn tap_bf16(
    gpu: &dyn GpuBackend,
    ptr: DevicePtr,
    layer: usize,
    tag: &str,
    n_elements: usize,
    stream: u64,
)
Expand description

Tap a BF16 buffer (the embedding, a block output) the same way.