Browse Source

!25921 Fix: cpu real_div support fp16.

Merge pull request !25921 from hezhenhao1/add_fp16
tags/v1.6.0
i-robot Gitee 4 years ago
parent
commit
fb155743a7
3 changed files with 6 additions and 1 deletions
  1. +4
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h
  2. +1
    -0
      mindspore/ops/_op_impl/cpu/real_div.py
  3. +1
    -1
      tests/st/ops/cpu/test_realdiv_op.py

+ 4
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.h View File

@@ -127,6 +127,10 @@ MS_REG_CPU_KERNEL_T(
MS_REG_CPU_KERNEL_T(
RealDiv, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt32),
ArithmeticCPUKernel, int32_t);
MS_REG_CPU_KERNEL_T(
RealDiv,
KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFloat16),
ArithmeticCPUKernel, float16);
MS_REG_CPU_KERNEL_T(
RealDiv,
KernelAttr().AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32),


+ 1
- 0
mindspore/ops/_op_impl/cpu/real_div.py View File

@@ -23,6 +23,7 @@ real_div_op_info = CpuRegOp("RealDiv") \
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
.dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.I64_Default) \
.dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.F64_Default) \
.get_op_info()




+ 1
- 1
tests/st/ops/cpu/test_realdiv_op.py View File

@@ -34,7 +34,7 @@ class NetRealDiv(nn.Cell):
@pytest.mark.level0
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard
@pytest.mark.parametrize('dtype', [np.float32, np.float64])
@pytest.mark.parametrize('dtype', [np.float32, np.float64, np.float16])
def test_real_div(dtype):
"""
Feature: ALL To ALL


Loading…
Cancel
Save