From 8eb54e18f97e5be53e53c01e79c22741dc16a7e0 Mon Sep 17 00:00:00 2001 From: liutuo Date: Tue, 19 Jun 2018 19:17:03 +0800 Subject: [PATCH] fix reduce mean half bug --- mace/kernels/opencl/reduce_mean_opencl.cc | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/mace/kernels/opencl/reduce_mean_opencl.cc b/mace/kernels/opencl/reduce_mean_opencl.cc index 8d47e4df..7930cc60 100644 --- a/mace/kernels/opencl/reduce_mean_opencl.cc +++ b/mace/kernels/opencl/reduce_mean_opencl.cc @@ -42,15 +42,8 @@ MaceStatus ReduceMeanFunctor::operator()( std::set built_options; std::string kernel_name = MACE_OBFUSCATE_SYMBOL("reduce_mean"); built_options.emplace("-Dreduce_mean=" + kernel_name); - - if (input->dtype() == output->dtype()) { - built_options.emplace("-DDATA_TYPE=" + DtToCLDt(dt)); - built_options.emplace("-DCMD_DATA_TYPE=" + DtToCLCMDDt(dt)); - built_options.emplace(dt == DT_HALF ? "-DFP16" : ""); - } else { - built_options.emplace("-DDATA_TYPE=" + DtToUpstreamCLDt(dt)); - built_options.emplace("-DCMD_DATA_TYPE=" + DtToUpstreamCLCMDDt(dt)); - } + built_options.emplace("-DDATA_TYPE=" + DtToUpstreamCLDt(dt)); + built_options.emplace("-DCMD_DATA_TYPE=" + DtToUpstreamCLCMDDt(dt)); if (runtime->gpu_type() != GPUType::QUALCOMM_ADRENO) { built_options.emplace("-DNON_QUALCOMM_ADRENO"); } -- GitLab