From c0ebc36c7843c7aab29b35a34037950f78390019 Mon Sep 17 00:00:00 2001 From: Peilin Wang Date: Sat, 20 Mar 2021 01:53:20 -0400 Subject: [PATCH] change cumsum op python to allow for float64 input --- mindspore/ops/operations/math_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/math_ops.py b/mindspore/ops/operations/math_ops.py index f1adace6d4..b9ff708874 100644 --- a/mindspore/ops/operations/math_ops.py +++ b/mindspore/ops/operations/math_ops.py @@ -972,7 +972,7 @@ class CumSum(PrimitiveWithInfer): if axis['value'] is None: raise ValueError(f"For {self.name}, axis must be const.") validator.check_value_type('axis', axis['value'], [int], cls_name) - valid_dtypes = [mstype.uint8, mstype.int8, mstype.int32, mstype.float16, mstype.float32] + valid_dtypes = [mstype.uint8, mstype.int8, mstype.int32, mstype.float16, mstype.float32, mstype.float64] validator.check_tensor_dtype_valid('x', x['dtype'], valid_dtypes, cls_name) return {'shape': x_shp, 'dtype': x['dtype'],