update api

PaddlePaddle · Apr 25, 2024 · 4fc88be · 4fc88be
1 parent be357d4
commit 4fc88be
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/nn/functional/__init__.py b/python/paddle/nn/functional/__init__.py
@@ -87,6 +87,8 @@
 )
 from .flash_attention import (
     flash_attention_with_sparse_mask,
+    flash_attn_qkvpacked,
+    flash_attn_varlen_qkvpacked,
     scaled_dot_product_attention,
     sdp_kernel,  # noqa: F401
 )
@@ -279,5 +281,7 @@
     'gaussian_nll_loss',
     'scaled_dot_product_attention',
     'flash_attention_with_sparse_mask',
+    'flash_attn_qkvpacked',
+    'flash_attn_varlen_qkvpacked',
     'group_norm',
 ]