Revert "fixed flaky test issue for test_operator_gpu.test_depthwise_c…

…onvolution (apache#12402)" (apache#12441) This reverts commit 58560f6.
anirudh2290 · Sep 19, 2018 · 1d8022a · 1d8022a
1 parent 02b838d
commit 1d8022a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -1616,6 +1616,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")
 @with_seed()
 def test_depthwise_convolution():
     for dim in [1,2]:
@@ -1649,15 +1650,15 @@ def test_depthwise_convolution():
                             exe2 = y2.simple_bind(mx.cpu(), 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.float32(np.random.normal(size=arr1.shape))
+                                arr1[:] = np.random.normal(size=arr1.shape)
                                 arr2[:] = arr1
                             exe1.forward(is_train=True)
                             exe1.backward(exe1.outputs[0])
                             exe2.forward(is_train=True)
                             exe2.backward(exe2.outputs[0])
 
                             for arr1, arr2 in zip(exe1.outputs + exe1.grad_arrays, exe2.outputs + exe2.grad_arrays):
-                                np.testing.assert_allclose(arr1.asnumpy(), arr2.asnumpy(), rtol=1e-2, atol=1e-3)
+                                np.testing.assert_allclose(arr1.asnumpy(), arr2.asnumpy(), rtol=1e-3, atol=1e-3)
 
 def gen_broadcast_data(idx):
     # Manually set test cases