[ROCM] fix native attention function call (#13650)

This commit is contained in:
Gordon Wong authored and GitHub committed 2025-02-21 22:07:04 -08:00
1 parent 68d535ef44
commit 68d630a0c7
1 file changed
-1
@@ -717,7 +717,6 @@ class ROCmFlashAttentionImpl(AttentionImpl):
self.num_heads,
self.head_size,
self.scale,
causal_mask,
attn_masks,
)
else: