Skip to content

Commit 8036db4

Browse files
committed
[XPU] support flashmask_attention forward
1 parent 389ba80 commit 8036db4

File tree

4 files changed

+486
-94
lines changed

4 files changed

+486
-94
lines changed

cmake/external/xpu.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ set(XPU_XFA_LIB_NAME "libxpu_flash_attention.so")
3030
set(XPU_XPUDNN_LIB_NAME "libxpu_dnn.so")
3131

3232
if(NOT DEFINED XPU_XHPC_BASE_DATE)
33-
set(XPU_XHPC_BASE_DATE "dev/20250220")
33+
set(XPU_XHPC_BASE_DATE "dev/20250310")
3434
endif()
3535
set(XPU_XCCL_BASE_VERSION "3.0.2.3") # For XRE5
3636
if(NOT DEFINED XPU_XFT_BASE_VERSION)

paddle/phi/backends/xpu/xpu3_op_list.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -604,6 +604,8 @@ XPUOpMap& get_kl3_ops() {
604604
XPUKernelSet({phi::DataType::BFLOAT16,
605605
phi::DataType::FLOAT32,
606606
phi::DataType::FLOAT16})},
607+
{"flashmask_attention",
608+
XPUKernelSet({phi::DataType::BFLOAT16, phi::DataType::FLOAT16})},
607609
{"flash_attn_unpadded",
608610
XPUKernelSet({phi::DataType::BFLOAT16,
609611
phi::DataType::FLOAT32,

0 commit comments

Comments
 (0)