Fix MXNDArrayGetData

apache · Sep 26, 2019 · d21136e · d21136e
1 parent c5007ea
commit d21136e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
@@ -628,11 +628,10 @@ int MXNDArrayGetData(NDArrayHandle handle,
   API_BEGIN();
   NDArray *arr = static_cast<NDArray*>(handle);
 #if MXNET_USE_MKLDNN == 1
-  NDArray temp = *arr;
   if (arr->IsMKLDNNData()) {
-    temp = arr->Reorder2Default();
-    arr = &temp;
+    arr->Reorder2DefaultAsync();
   }
+  arr->WaitToRead();
 #endif
   if (!arr->is_none()) {
     *out_pdata = arr->data().dptr_;