diff --git a/mace/kernels/opencl/reduce_mean_opencl.cc b/mace/kernels/opencl/reduce_mean_opencl.cc index 8d47e4df610cc86c9aa68aa6a2ef9a1b6a0ae7b2..7930cc6047459ea8f7841c70eef43dd349385495 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"); }