diff --git a/paddle/fluid/platform/device/xpu/xpu_op_kpfirst_list.h b/paddle/fluid/platform/device/xpu/xpu_op_kpfirst_list.h index 452f388f03dcf9643fdff89edb27578c732c490c..ff44fa0b77201da92a96aaaed7651f221e772da3 100644 --- a/paddle/fluid/platform/device/xpu/xpu_op_kpfirst_list.h +++ b/paddle/fluid/platform/device/xpu/xpu_op_kpfirst_list.h @@ -100,19 +100,28 @@ XPUOpMap& get_kp_ops() { {"equal", XPUKernelSet({pOpKernelType(vartype::INT32, XPUPlace())})}, {"not_equal", XPUKernelSet({pOpKernelType(vartype::INT32, XPUPlace())})}, // reduce op - {"reduce_mean", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_max", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_min", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_sum", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_prod", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_all", XPUKernelSet({pOpKernelType(vartype::BOOL, XPUPlace())})}, - {"reduce_any", XPUKernelSet({pOpKernelType(vartype::BOOL, XPUPlace())})}, + // {"reduce_mean", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_max", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_min", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_sum", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_prod", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_all", XPUKernelSet({pOpKernelType(vartype::BOOL, + // XPUPlace())})}, + // {"reduce_any", XPUKernelSet({pOpKernelType(vartype::BOOL, + // XPUPlace())})}, + // {"reduce_amax", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, + // {"reduce_amin", XPUKernelSet({pOpKernelType(vartype::FP32, + // XPUPlace())})}, {"pull_box_sparse", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"push_box_sparse", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_amax", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, - {"reduce_amin", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"c_sync_calc_stream", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"c_sync_comm_stream",