!1222 GPU kernel add type validation

Merge pull request !1222 from VectorSL/floatstatus-typecheck

!1222 GPU kernel add type validation
Merge pull request !1222 from VectorSL/floatstatus-typecheck
699d0c10 · mindspore-ci-bot · Gitee · 1b7cdc4c · 2b511990 · 699d0c10
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

mindspore/ops/operations/math_ops.py mindspore/ops/operations/math_ops.py +1 -0

未找到文件。
--- a/mindspore/ops/operations/math_ops.py
+++ b/mindspore/ops/operations/math_ops.py
@@ -1712,6 +1712,7 @@ class FloatStatus(PrimitiveWithInfer):
        return [1]

    def infer_dtype(self, x_dtype):
+        validator.check_tensor_type_same({'x': x_dtype}, [mstype.float32, mstype.float16], self.name)
        return x_dtype

 class NPUAllocFloatStatus(PrimitiveWithInfer):