Browse Source
!6542 GPU equal add types
Merge pull request !6542 from VectorSL/equal-typetags/v1.0.0
mindspore-ci-bot
Gitee
5 years ago
3 changed files with 6 additions and 0 deletions
Unified View
Diff Options
-
+2 -0mindspore/ops/_op_impl/akg/gpu/equal.py
-
+2 -0mindspore/ops/_op_impl/akg/gpu/greater_equal.py
-
+2 -0mindspore/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") \ | .input(1, "y") \ | ||||
| .output(0, "output") \ | .output(0, "output") \ | ||||
| .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \ | .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.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.I32_Default, DataType.I32_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.I16_Default, DataType.I16_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.U8_Default, DataType.U8_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) \ | .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default, DataType.BOOL_Default) \ | ||||
| .get_op_info() | .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") \ | .output(0, "output") \ | ||||
| .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \ | .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.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.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.I16_Default, DataType.I16_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.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \ | |||||
| .get_op_info() | .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") \ | .output(0, "output") \ | ||||
| .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \ | .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.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.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.I16_Default, DataType.I16_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.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \ | ||||
| .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \ | |||||
| .get_op_info() | .get_op_info() | ||||