Browse Source

!2685 GPU mul support int

Merge pull request !2685 from VectorSL/r0.5
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
acbbe52984
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