[XPU] fix int64 index bug of masked_select (#72762)

cqulilujia · web-flow · commit 1f476d751ce5 · 2025-05-19T10:36:50.000+08:00
diff --git a/paddle/phi/kernels/xpu/masked_select_kernel.cc b/paddle/phi/kernels/xpu/masked_select_kernel.cc
@@ -55,7 +55,7 @@ void MaskedSelectKernel(const Context& dev_ctx,
                      static_cast<void*>(&out_size_cpu),
                      mask.place(),
                      static_cast<void*>(out_size),
-                     sizeof(int32_t));
+                     sizeof(int64_t));
   if (std::getenv("XPUSIM_SKIP_RUN") &&
       std::strcmp(std::getenv("XPUSIM_SKIP_RUN"), "1") == 0) {
     VLOG(3) << "WARNING: In the simulator mode, the variable out_size_cpu "
diff --git a/paddle/phi/kernels/xpu/warpctc_kernel.cc b/paddle/phi/kernels/xpu/warpctc_kernel.cc
@@ -115,7 +115,7 @@ void WarpctcKernel(const Context& dev_ctx,
   dev_ctx.template Alloc<T>(warpctcgrad);
   T* warpctcgrad_data = warpctcgrad->data<T>();
 
-  int sm_workspace, lm_workspace;
+  int64_t sm_workspace, lm_workspace;
   int64_t max_S = 2 * max_target_seq_length + 1;
   if (warpctcgrad_data == nullptr) {
     sm_workspace = sizeof(T) * sequence_width +
