From 31b73346fb7e0264ca1cdf49e08e78ba72f5a397 Mon Sep 17 00:00:00 2001 From: niuliling123 <51102941+niuliling123@users.noreply.github.com> Date: Fri, 17 Jun 2022 10:06:38 +0800 Subject: [PATCH] delete XPU reduce for yolov3 (#43591) --- .../platform/device/xpu/xpu_op_kpfirst_list.h | 27 ++++++++++++------- 1 file changed, 18 insertions(+), 9 deletions(-) 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 452f388f03d..ff44fa0b772 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", -- GitLab