cherry-pick from (PaddlePaddle#56066): fix codestyle

hitywt · Oct 26, 2023 · c57972a · c57972a
1 parent 8e90e92
commit c57972a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/paddle/phi/api/yaml/ops.yaml b/paddle/phi/api/yaml/ops.yaml
@@ -925,7 +925,6 @@
   kernel :
     func : flash_attn
     data_type : q
-  intermediate : softmax_lse, seed_offset
   backward : flash_attn_grad
 
 - op : flash_attn_unpadded
@@ -950,7 +949,6 @@
   kernel :
     func : flash_attn_v1
     data_type : q
-  intermediate : softmax_lse, seed_offset
   backward : flash_attn_v1_grad
 
 - op : flash_attn_v1_unpadded

diff --git a/python/paddle/nn/functional/flash_attention.py b/python/paddle/nn/functional/flash_attention.py
@@ -201,10 +201,7 @@ def flash_attention(
     if sdp_func_name == "flash_attn":
         if in_dynamic_mode():
             if g_use_flash_attn_v1:
-                (
-                    result_attention,
-                    result_softmax,
-                ) = _C_ops.flash_attn_v1(
+                (result_attention, result_softmax, _, _) = _C_ops.flash_attn_v1(
                     query,
                     key,
                     value,
@@ -214,10 +211,7 @@ def flash_attention(
                     not training,
                 )
             else:
-                (
-                    result_attention,
-                    result_softmax,
-                ) = _C_ops.flash_attn(
+                (result_attention, result_softmax, _, _) = _C_ops.flash_attn(
                     query,
                     key,
                     value,