Browse Source

!6542 GPU equal add types

Merge pull request !6542 from VectorSL/equal-type
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
7e9a16de3b
3 changed files with 6 additions and 0 deletions
  1. +2
    -0
      mindspore/ops/_op_impl/akg/gpu/equal.py
  2. +2
    -0
      mindspore/ops/_op_impl/akg/gpu/greater_equal.py
  3. +2
    -0
      mindspore/ops/_op_impl/akg/gpu/lessequal.py

+ 2
- 0
mindspore/ops/_op_impl/akg/gpu/equal.py View File

@@ -21,10 +21,12 @@ equal_op_info = AkgGpuRegOp("Equal") \
.input(1, "y") \
.output(0, "output") \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I16_Default, DataType.I16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \
.dtype_format(DataType.BOOL_Default, DataType.BOOL_Default, DataType.BOOL_Default) \
.get_op_info()



+ 2
- 0
mindspore/ops/_op_impl/akg/gpu/greater_equal.py View File

@@ -22,9 +22,11 @@ greater_equal_op_info = AkgGpuRegOp("GreaterEqual") \
.output(0, "output") \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I16_Default, DataType.I16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \
.get_op_info()




+ 2
- 0
mindspore/ops/_op_impl/akg/gpu/lessequal.py View File

@@ -22,9 +22,11 @@ lessequal_op_info = AkgGpuRegOp("LessEqual") \
.output(0, "output") \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I16_Default, DataType.I16_Default, DataType.BOOL_Default) \
.dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \
.dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \
.get_op_info()




Loading…
Cancel
Save