From 40078103c480ba9a6cf59deb7fdba951c69b62f2 Mon Sep 17 00:00:00 2001
From: TTerror <tangzhiyi11@users.noreply.github.com>
Date: Wed, 5 Jan 2022 17:10:52 +0800
Subject: [PATCH] update masked_select_op for kunlun (#38678)

---
 cmake/external/xpu.cmake                       | 2 +-
 paddle/fluid/operators/masked_select_op_xpu.cc | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmake/external/xpu.cmake b/cmake/external/xpu.cmake
index 588ba0bfe86..c7a6f04b5f4 100644
--- a/cmake/external/xpu.cmake
+++ b/cmake/external/xpu.cmake
@@ -36,7 +36,7 @@ ENDIF()
 
 if(NOT DEFINED XPU_BASE_URL)
   SET(XPU_BASE_URL_WITHOUT_DATE "https://baidu-kunlun-product.cdn.bcebos.com/KL-SDK/klsdk-dev")
-  SET(XPU_BASE_URL "${XPU_BASE_URL_WITHOUT_DATE}/20211228")
+  SET(XPU_BASE_URL "${XPU_BASE_URL_WITHOUT_DATE}/20220104")
 else()
   SET(XPU_BASE_URL "${XPU_BASE_URL}")
 endif()
diff --git a/paddle/fluid/operators/masked_select_op_xpu.cc b/paddle/fluid/operators/masked_select_op_xpu.cc
index aafc2510a8c..c575f133b15 100644
--- a/paddle/fluid/operators/masked_select_op_xpu.cc
+++ b/paddle/fluid/operators/masked_select_op_xpu.cc
@@ -57,7 +57,7 @@ class MaskedSelectXPUKernel : public framework::OpKernel<T> {
 
     PADDLE_ENFORCE_XPU_SUCCESS(
         xpu::masked_select(dev_ctx.x_context(), input_data, mask_data, out_data,
-                           input_shape, mask_shape));
+                           input_shape, mask_shape, out_size_cpu));
   }
 };
 
-- 
GitLab