pub fn launch<R: Runtime, AP: AttentionPrecision>(
strategy: &Strategy,
client: &ComputeClient<R::Server>,
query: TensorHandle<R, QG<AP>>,
key: TensorHandle<R, KG<AP>>,
value: TensorHandle<R, VG<AP>>,
mask: Option<TensorHandle<R, MSK<AP>>>,
out: TensorHandle<R, OG<AP>>,
) -> Result<(), AttentionSetupError>