Fix MXNDArrayGetData (#16289)

apache · Sep 27, 2019 · f52ddfd · f52ddfd
1 parent c69c8bf
commit f52ddfd
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()) {