Skip to content

Commit

Permalink
fix test_depthwise_convoltuion for occasional CI failures (apache#14016)
Browse files Browse the repository at this point in the history
* keeping same contexts for comparison

* enabling test

* testing default context

* Revert "testing default context"

This reverts commit 1f95d02.

* Disabling test due to CI failure on MKL-DNN
  • Loading branch information
mseth10 authored and haohuw committed Jun 23, 2019
1 parent 48aa673 commit ee63ebd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python/unittest/test_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -1631,7 +1631,7 @@ def test_convolution_grouping():
np.testing.assert_allclose(arr1.asnumpy(), arr2.asnumpy(), rtol=1e-3, atol=1e-3)


@unittest.skip("Flaky test https://github.com/apache/incubator-mxnet/issues/12203")
@unittest.skip("Flaky test https://github.com/apache/incubator-mxnet/issues/14052")
@with_seed()
def test_depthwise_convolution():
for dim in [1,2]:
Expand Down Expand Up @@ -1662,7 +1662,7 @@ def test_depthwise_convolution():

dev = default_context()
exe1 = y1.simple_bind(dev, x=shape)
exe2 = y2.simple_bind(mx.cpu(), x=shape, w=(num_filter, shape[1]//num_group)+kernel,
exe2 = y2.simple_bind(dev, x=shape, w=(num_filter, shape[1]//num_group)+kernel,
b=(num_filter,))
for arr1, arr2 in zip(exe1.arg_arrays, exe2.arg_arrays):
arr1[:] = np.random.normal(size=arr1.shape)
Expand Down

0 comments on commit ee63ebd

Please sign in to comment.