[incubator-mxnet] branch master updated: add flaky to norm (#20091)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 4552d4f  add flaky to norm (#20091)
4552d4f is described below

commit 4552d4f976deb0d695aadcae0bdf9b570829f89e
Author: barry-jin <69...@users.noreply.github.com>
AuthorDate: Sat Apr 3 13:48:29 2021 -0700

    add flaky to norm (#20091)
---
 tests/python/unittest/test_operator.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
index 88bd4c3..31abf87 100644
--- a/tests/python/unittest/test_operator.py
+++ b/tests/python/unittest/test_operator.py
@@ -3410,6 +3410,7 @@ def check_layer_normalization(in_shape, axis, eps, dtype=np.float32,
                             gt_beta_grad + init_beta_grad, backward_check_eps, backward_check_eps)
 
 
+@pytest.mark.flaky
 def test_norm():
     try:
         import scipy
@@ -3490,6 +3491,7 @@ def test_norm():
     (np.float32, 1E-3, 1E-3, [(10, 6, 5), (10, 10), (128 * 32, 512)], [True, True, False]),
     (np.float64, 1E-4, 1E-4, [(10, 6, 5), (10, 10), (128 * 32, 512)], [True, True, False])
 ])
+@pytest.mark.flaky
 def test_layer_norm(enforce_safe_acc, dtype, forward_check_eps, backward_check_eps,
                     in_shape_l, finite_grad_check_l):
     with environment('MXNET_SAFE_ACCUMULATION', enforce_safe_acc):