only reorder if mkldnn

apache · Aug 7, 2018 · ec1d179 · ec1d179
1 parent c0e639a
commit ec1d179
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_base-inl.h b/src/operator/nn/mkldnn/mkldnn_base-inl.h
@@ -353,8 +353,12 @@ static inline void InvalidateOutputs(const std::vector<NDArray> &arrs,
 
 static inline std::vector<NDArray> CreateDefaultInputs(const std::vector<NDArray> &arrs) {
   std::vector<NDArray> buffer(arrs.size());
-  for (size_t i = 0; i < arrs.size(); ++i)
-    buffer[i] = arrs[i].Reorder2Default();
+  for (size_t i = 0; i < arrs.size(); ++i) {
+    if (arrs[i].IsMKLDNNData()) {
+      buffer[i] = arrs[i].Reorder2Default();
+    }
+    buffer[i] = arrs[i];
+  }
   return buffer;
 }