launch_ref

cubecl_attention

Function launch_ref 

Source 
pub fn launch_ref<R: Runtime, AP: AttentionPrecision>(
    strategy: &Strategy,
    client: &ComputeClient<R::Server>,
    query: &TensorHandleRef<'_, R>,
    key: &TensorHandleRef<'_, R>,
    value: &TensorHandleRef<'_, R>,
    mask: &Option<TensorHandleRef<'_, R>>,
    out: &TensorHandleRef<'_, R>,
) -> Result<(), AttentionSetupError>