[incubator-mxnet] branch master updated: Fix MKLDNNSum cpp test failure (#12080)

[an...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

anirudh2290 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new ab92fd8  Fix MKLDNNSum cpp test failure (#12080)
ab92fd8 is described below

commit ab92fd80b10805fd103217df0569513289db4625
Author: Jin Huang <34...@users.noreply.github.com>
AuthorDate: Thu Aug 16 12:49:38 2018 +0800

    Fix MKLDNNSum cpp test failure (#12080)
---
 tests/cpp/operator/mkldnn.cc | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/cpp/operator/mkldnn.cc b/tests/cpp/operator/mkldnn.cc
index 6cb1400..59bd3a5 100644
--- a/tests/cpp/operator/mkldnn.cc
+++ b/tests/cpp/operator/mkldnn.cc
@@ -1268,8 +1268,6 @@ TEST(IMPERATIVE, PoolingOp) {
   }
 }
 
-// Disabling Flaky Test. Tracked at https://github.com/apache/incubator-mxnet/issues/11998
-/*
 TEST(MKLDNN_BASE, MKLDNNSum) {
   std::vector<NDArrayAttrs> in_arrs = GetTestInputArrays();
   std::vector<NDArrayAttrs> in_arrs2 = GetTestInputArrays(true);
@@ -1310,6 +1308,7 @@ TEST(MKLDNN_BASE, MKLDNNSum) {
     auto input_mem = in_arr.arr.GetMKLDNNData();
     auto input_mem2 = in_arr2.arr.GetMKLDNNData();
     NDArrayAttrs orig_arr(in_arr.arr.Copy(in_arr.arr.ctx()), "In Place Copy");
+    orig_arr.arr.WaitToRead();
     PrintVerifyMsg(orig_arr, in_arr);
     InitMKLDNNArray(&orig_arr.arr, input_mem->get_primitive_desc());
     orig_arr.arr.CopyFrom(*input_mem);
@@ -1318,7 +1317,6 @@ TEST(MKLDNN_BASE, MKLDNNSum) {
     VerifySumResult({&orig_arr.arr, &in_arr2.arr}, {&in_arr.arr});
   }
 }
-*/
 
 TEST(MKLDNN_BASE, CreateMKLDNNMem) {
   std::vector<NDArrayAttrs> in_arrs = GetTestInputArrays();