SDPA backend priority (#9299)

This commit is contained in:
contentis
2025-08-13 20:53:27 +02:00
committed by GitHub
parent afa0a45206
commit 3da5a07510
4 changed files with 17 additions and 4 deletions

View File

@@ -178,7 +178,7 @@ class FourierEmbedder(nn.Module):
class CrossAttentionProcessor:
def __call__(self, attn, q, k, v):
out = F.scaled_dot_product_attention(q, k, v)
out = ops.scaled_dot_product_attention(q, k, v)
return out