From 79c4312e892f99e19672e9a04e6a11d6a98b7c9b Mon Sep 17 00:00:00 2001 From: candanzg Date: Sat, 18 Apr 2020 16:33:30 +0800 Subject: [PATCH] NotEqual op auto cast Signed-off-by: candanzg --- tests/ut/python/ops/test_ops.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/ut/python/ops/test_ops.py b/tests/ut/python/ops/test_ops.py index 078ada84065..eee5080a6cc 100755 --- a/tests/ut/python/ops/test_ops.py +++ b/tests/ut/python/ops/test_ops.py @@ -401,6 +401,11 @@ test_case_math_ops = [ 'block': P.NotEqual(), 'desc_inputs': [[4, 1], [2, 3, 4, 5]], 'desc_bprop': [Tensor(np.ones((2, 3, 4, 5), np.bool_))]}), + ('NotEqual_0', { + 'block': P.NotEqual(), + 'desc_inputs': [ 1, [2, 3, 4, 5]], + 'desc_bprop': [Tensor(np.ones((2, 3, 4, 5), np.bool_))], + 'skip': ['backward']}), ('Greater', { 'block': P.Greater(), 'desc_inputs': [[2, 3, 4, 1], [4, 5]],