pub fn dispatch_sqrt_f32(
encoder: &mut CommandEncoder,
registry: &mut KernelRegistry,
device: &DeviceRef,
input: &MlxBuffer,
output: &MlxBuffer,
params: &MlxBuffer,
) -> Result<()>pub fn dispatch_sqrt_f32(
encoder: &mut CommandEncoder,
registry: &mut KernelRegistry,
device: &DeviceRef,
input: &MlxBuffer,
output: &MlxBuffer,
params: &MlxBuffer,
) -> Result<()>