!2684 GPU add mul support int

Merge pull request !2684 from VectorSL/mul

!2684 GPU add mul support int
Merge pull request !2684 from VectorSL/mul
4e1d9ede · mindspore-ci-bot · Gitee · f900a224 · b854ae7e · 4e1d9ede
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

mindspore/ccsrc/kernel/gpu/math/broadcast_gpu_kernel.cc mindspore/ccsrc/kernel/gpu/math/broadcast_gpu_kernel.cc +3 -0

未找到文件。
--- a/mindspore/ccsrc/kernel/gpu/math/broadcast_gpu_kernel.cc
+++ b/mindspore/ccsrc/kernel/gpu/math/broadcast_gpu_kernel.cc
@@ -96,5 +96,8 @@ MS_REG_GPU_KERNEL_TWO(
 MS_REG_GPU_KERNEL_TWO(
  Maximum, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32),
  BroadcastOpGpuKernel, int, int)
+MS_REG_GPU_KERNEL_TWO(
+  Mul, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32),
+  BroadcastOpGpuKernel, int, int)
 }  // namespace kernel
 }  // namespace mindspore