Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Mark test_batchnorm_with_type as flaky (#19492)
Browse files Browse the repository at this point in the history
[2020-11-06T21:37:09.831Z]       # V2, 1D
[2020-11-06T21:37:09.831Z]       for fix_gamma, cudnn_off in itertools.product(bools, bools):
[2020-11-06T21:37:09.831Z]           sym = mx.sym.BatchNorm(name='norm', fix_gamma=fix_gamma, cudnn_off=cudnn_off)
[2020-11-06T21:37:09.831Z] >         check_consistency(sym, ctx_list_v2_1D)
[...]
[2020-11-06T21:37:09.831Z] E       AssertionError: 
[2020-11-06T21:37:09.831Z] E       Items are not equal:
[2020-11-06T21:37:09.831Z] E       Error 8317.657381 exceeds tolerance rtol=1.000000e-05, atol=1.000000e-20 (mismatch 50.000000%).
[2020-11-06T21:37:09.831Z] E       Location of maximum error: (0,), a=-0.00000000, b=-0.00000000
[2020-11-06T21:37:09.831Z] E        ACTUAL: array([ -0.        , -39.62720511])
[2020-11-06T21:37:09.831Z] E        DESIRED: array([ -0.        , -39.62720511])
  • Loading branch information
leezu authored Nov 8, 2020
1 parent abce30a commit 545e117
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/python/gpu/test_operator_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
# specific language governing permissions and limitations
# under the License.

from __future__ import print_function
import sys
import os
import time
Expand Down Expand Up @@ -372,6 +371,7 @@ def test_preloaded_multi_sgd():


@pytest.mark.serial
@pytest.mark.flaky
def test_batchnorm_with_type():
ctx_list_v2_2D = [
{'ctx': mx.cpu(0), 'norm_data': (5, 2, 5, 5), 'type_dict': {'norm_data': np.float32}},
Expand Down

0 comments on commit 545e117

Please sign in to comment.