From 2ddd34f8e735f40f6caa3a112a28e8b576a1b680 Mon Sep 17 00:00:00 2001 From: Sheng Zha Date: Mon, 5 Apr 2021 13:01:48 -0400 Subject: [PATCH] Revert "add flaky to norm (#20091)" This reverts commit 4552d4f976deb0d695aadcae0bdf9b570829f89e. --- tests/python/unittest/test_operator.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py index 31abf87d6e79..88bd4c3be04c 100644 --- a/tests/python/unittest/test_operator.py +++ b/tests/python/unittest/test_operator.py @@ -3410,7 +3410,6 @@ 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 @@ -3491,7 +3490,6 @@ 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):