fixed seed for mkldnn test (apache#17386)

* fix seed for mkldnn test * use fixed seed for mkldnn test
yajiedesign · Jan 25, 2020 · 47349ce · 47349ce
1 parent e1779f4
commit 47349ce
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/python/mkl/test_mkldnn.py b/tests/python/mkl/test_mkldnn.py
@@ -61,6 +61,7 @@ def get_tensors(args, shapes, ctx):
     except:  # pylint: disable=bare-except
         assert 0, "test_mkldnn_model exception in bind and execution"
 
+@with_seed(1234)
 def test_mkldnn_ndarray_slice():
     ctx = mx.cpu()
     net = gluon.nn.HybridSequential()
@@ -71,8 +72,9 @@ def test_mkldnn_ndarray_slice():
     y = net(x)
 
     # trigger computation on ndarray slice
-    assert_almost_equal(y[0].asnumpy()[0, 0, 0], np.array(0.3376348))
+    assert_almost_equal(y[0].asnumpy()[0, 0, 0], np.array(0.056331709))
 
+@with_seed(1234)
 def test_mkldnn_engine_threading():
     net = gluon.nn.HybridSequential()
     with net.name_scope():
@@ -94,8 +96,8 @@ def __getitem__(self, key):
     # below line triggers different execution thread
     for _ in loader:
         y = net(mx.nd.array(np.ones(X))).asnumpy()
-        # output should be 016711406 (non-mkldnn mode output)
-        assert_almost_equal(y[0, 0, 0, 0], np.array(0.016711406))
+        # output should be 056331709 (non-mkldnn mode output)
+        assert_almost_equal(y[0, 0, 0, 0], np.array(0.056331709))
         break
 
 @with_seed()