diff --git a/cmake/external/xpu.cmake b/cmake/external/xpu.cmake index 588ba0bfe86cd5d252b85d8373270fae1a79cf5e..c7a6f04b5f40a202d849d91d3d07a8bfb3ea7fff 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 aafc2510a8c44e7bd73261c7ef44227325a8e6b2..c575f133b15728ec2b8599329f6d50878d371aec 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 { 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)); } };