Function register 

pub fn register(registry: &mut KernelRegistry)

Register SDPA shader source with the given kernel registry.

This must be called before dispatching any SDPA operations.