From 7dddc31e91eb77448ab9026d2e055e736a91885f Mon Sep 17 00:00:00 2001 From: barry-jin Date: Thu, 25 Mar 2021 10:53:21 -0700 Subject: [PATCH] add flaky to norm --- 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 88bd4c3be04c..31abf87d6e79 100644 --- a/tests/python/unittest/test_operator.py +++ b/tests/python/unittest/test_operator.py @@ -3410,6 +3410,7 @@ def npy_layer_norm_grad(data, gamma, out_grad, axis, eps): 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 l2norm(input_data, axis=0, keepdims=True): (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):