From 98d2c94700509ae2ce2514b40e94a870ee06199c Mon Sep 17 00:00:00 2001 From: Peilin Wang Date: Wed, 3 Mar 2021 18:16:32 -0500 Subject: [PATCH] add float64 support to greater and less --- .../kernel_compiler/gpu/math/broadcast_gpu_kernel.cc | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_gpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_gpu_kernel.cc index 8d232afcd7..2801689dae 100644 --- 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)