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

Fix test_quantization to unblock CD #18307

Merged
leezu merged 1 commit into from
May 14, 2020
Merged

Fix test_quantization to unblock CD #18307

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion tests/python/quantization/test_quantization.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
import numpy as np
from mxnet.gluon.model_zoo import vision
from mxnet.test_utils import assert_almost_equal, assert_exception, rand_ndarray, rand_shape_nd, same, DummyIter
from common import with_seed
from common import with_seed, xfail_when_nonstandard_decimal_separator
from mxnet.module import Module
from mxnet.io import NDArrayIter
import unittest
Expand Down Expand Up @@ -870,6 +870,7 @@ def get_fp32_sym_with_multiple_outputs(length=1):
out_grad=False, preserve_shape=False, use_ignore=False, name='softmax')
return sym

@xfail_when_nonstandard_decimal_separator
@with_seed()
def test_quantize_model():
def check_quantize_model(qdtype):
Expand Down Expand Up @@ -1170,6 +1171,7 @@ def check_quantize_net(qdtype):
for qdtype in ['int8', 'uint8']:
check_quantize_net(qdtype)

@xfail_when_nonstandard_decimal_separator
@with_seed()
def test_quantize_sym_with_calib():
if is_test_for_native_cpu():
Expand Down