Function moe_zero_buffer_encode 

pub fn moe_zero_buffer_encode(
    encoder: &mut CommandEncoder,
    registry: &mut KernelRegistry,
    device: &DeviceRef,
    output: &MlxBuffer,
    n_elements: usize,
) -> Result<()>

Zero-initialize an f32 GPU buffer using the zero_buffer kernel.

This is useful for preparing an accumulator buffer before dispatching weighted accumulation passes.

Arguments

• encoder — Command encoder to record into.
• registry — Kernel registry for pipeline lookup.
• device — Metal device reference.
• output — f32 buffer to zero, must be at least n_elements * 4 bytes.
• n_elements — Number of f32 elements to zero.