Fix MXNDArrayGetData (apache#16289)

larroy · Sep 28, 2019 · 4761f93 · 4761f93
1 parent 3a5158c
commit 4761f93
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,10 +628,9 @@ 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()) {