Skip to main content

register

mlx_native::ops::kv_cache_copy

Function register 

Source 
pub fn register(registry: &mut KernelRegistry)
Expand description

Register KV cache copy shader source with the given kernel registry.