From 27400afe8438a99e738137e743908b055d904da8 Mon Sep 17 00:00:00 2001 From: KylinMoriarty Date: Mon, 15 Aug 2022 10:19:11 +0800 Subject: [PATCH] delete fp16 dtype of realdiv and scatternd --- .../ccsrc/plugin/device/cpu/kernel/arithmetic_cpu_kernel.cc | 4 +--- .../ccsrc/plugin/device/cpu/kernel/scatter_nd_cpu_kernel.cc | 2 -- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/arithmetic_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/arithmetic_cpu_kernel.cc index 52b42a979b6..1b4032bc419 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/arithmetic_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/arithmetic_cpu_kernel.cc @@ -892,9 +892,7 @@ static std::map}}}, {kRealDiv, - {{KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFloat16), - SpecializeArithFunc}, - {KernelAttr().AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32), + {{KernelAttr().AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32), SpecializeArithFunc}, {KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTypeFloat64), SpecializeArithFunc}, diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/scatter_nd_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/scatter_nd_cpu_kernel.cc index 48d535b7d95..e44ec54e0d3 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/scatter_nd_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/scatter_nd_cpu_kernel.cc @@ -201,7 +201,6 @@ void ScatterNdCpuKernelMod::Check(const CNodePtr &kernel_node) const { std::vector> ScatterNdCpuKernelMod::func_list_ = { DTYPE_REGISTER_ATTR(kNumberTypeFloat64, kNumberTypeFloat64, double), DTYPE_REGISTER_ATTR(kNumberTypeFloat32, kNumberTypeFloat32, float), - DTYPE_REGISTER_ATTR(kNumberTypeFloat16, kNumberTypeFloat16, float16), DTYPE_REGISTER_ATTR(kNumberTypeInt64, kNumberTypeInt64, int64_t), DTYPE_REGISTER_ATTR(kNumberTypeInt32, kNumberTypeInt32, int32_t), DTYPE_REGISTER_ATTR(kNumberTypeInt16, kNumberTypeInt16, int16_t), @@ -212,7 +211,6 @@ std::vector> Scatter DTYPE_REGISTER_ATTR(kNumberTypeUInt8, kNumberTypeUInt8, uint8_t), DTYPE_REGISTER(kNumberTypeFloat64, kNumberTypeInt64, kNumberTypeFloat64, double), DTYPE_REGISTER(kNumberTypeFloat32, kNumberTypeInt64, kNumberTypeFloat32, float), - DTYPE_REGISTER(kNumberTypeFloat16, kNumberTypeInt64, kNumberTypeFloat16, float16), DTYPE_REGISTER(kNumberTypeInt64, kNumberTypeInt64, kNumberTypeInt64, int64_t), DTYPE_REGISTER(kNumberTypeInt32, kNumberTypeInt64, kNumberTypeInt32, int32_t), DTYPE_REGISTER(kNumberTypeInt16, kNumberTypeInt64, kNumberTypeInt16, int16_t),