gpu mul support int

b854ae7e · VectorSL · e83c5630 · b854ae7e
隐藏空白更改
内联并排

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