Browse Source

!2684 GPU add mul support int

Merge pull request !2684 from VectorSL/mul
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
4e1d9ede5b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/kernel/gpu/math/broadcast_gpu_kernel.cc

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

@@ -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

Loading…
Cancel
Save