[XPU] Update XHPC to 20251014 and add some dim check in FlashAttnKernel of xpu. (PaddlePaddle#75872)

Author: ZibinGuo

cmake/external/xpu.cmake

@@ -34,9 +34,9 @@ set(XPU_FFT_LIB_NAME "libcufft.so")
 add_compile_definitions(XPUAPI_NOT_INCLUDE_DEPRECATED)
 
 if(NOT DEFINED XPU_XHPC_BASE_DATE)
-  set(XPU_XHPC_BASE_DATE "dev/20251010")
+  set(XPU_XHPC_BASE_DATE "dev/20251014")
 endif()
-set(XPU_XCCL_BASE_VERSION "3.0.3.3") # For XRE5
+set(XPU_XCCL_BASE_VERSION "3.0.3.4") # For XRE5
 if(NOT DEFINED XPU_XFT_BASE_VERSION)
   set(XPU_XFT_BASE_VERSION "20250507/xpu3")
 endif()

paddle/phi/kernels/xpu/flash_attn_kernel.cc

@@ -462,6 +462,21 @@ void FlashAttnKernel(const Context& dev_ctx,
                     common::errors::InvalidArgument(
                         "flash_attn receive input with dim "
                         "[batch_size, seq_len, num_heads, head_dim]"));
+  PADDLE_ENFORCE_EQ(k.dims().size(),
+                    4,
+                    common::errors::InvalidArgument(
+                        "flash_attn receive input with dim "
+                        "[batch_size, seq_len, num_heads, head_dim]"));
+  PADDLE_ENFORCE_EQ(v.dims().size(),
+                    4,
+                    common::errors::InvalidArgument(
+                        "flash_attn receive input with dim "
+                        "[batch_size, seq_len, num_heads, head_dim]"));
+  PADDLE_ENFORCE_EQ(out->dims().size(),
+                    4,
+                    common::errors::InvalidArgument(
+                        "flash_attn receive input with dim "
+                        "[batch_size, seq_len, num_heads, head_dim]"));
 
   const int64_t batch_size = dims[0];
   const int64_t seqlen_q = dims[1];