!12845 Add float64 type support to Greater and Less

From: @peilin-wang Reviewed-by: Signed-off-by:
4 years ago · 9c1e73a5b9
parent ea8db7baf4 98d2c94700
commit 9c1e73a5b9
1 changed files with 7 additions and 0 deletions
--- a/mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_gpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_gpu_kernel.cc
@ -19,6 +19,13 @@
 namespace mindspore {
 namespace kernel {
 // fp64
+MS_REG_GPU_KERNEL_ONE(
+  Greater,
+  KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTypeBool),
+  BroadcastOpGpuKernel, double)
+MS_REG_GPU_KERNEL_ONE(
+  Less, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTypeBool),
+  BroadcastOpGpuKernel, double)
 MS_REG_GPU_KERNEL_ONE(
  Add, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberTypeFloat64),
  BroadcastOpGpuKernel, double)