fix error w/ flash-attn v2

facebookresearch · Aug 4, 2023 · 038a4a5 · 038a4a5
1 parent 52ca08d
commit 038a4a5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xformers/ops/fmha/flash.py b/xformers/ops/fmha/flash.py
@@ -67,6 +67,7 @@ def _flash_fwd(
             out_padded,
             softmax_lse,
             p,
+            _, # rng_state
         ) = _C_flashattention.varlen_fwd(
             query,
             key,