!2685 GPU mul support int

Merge pull request !2685 from VectorSL/r0.5

!2685 GPU mul support int
Merge pull request !2685 from VectorSL/r0.5
acbbe529 · mindspore-ci-bot · Gitee · f8b608cb · 79ec5f73 · acbbe529
隐藏空白更改
内联并排

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