Reorder2Default: return directly for default format (#10810)

apache · May 4, 2018 · ab6a25e · ab6a25e
1 parent e49fdae
commit ab6a25e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ndarray/ndarray.cc b/src/ndarray/ndarray.cc
@@ -348,7 +348,8 @@ void NDArray::Chunk::Reorder2Default() {
     return;
 
   mkldnn_memory_format_t format = mkl_mem_->GetDefaultFormat();
-  CHECK_NE(format, mkl_mem_->GetFormat());
+  if (format ==  mkl_mem_->GetFormat())
+    return;
 
   mkldnn::memory::primitive_desc def_pd = mkl_mem_->GetPrimitiveDesc(format);
   mkldnn_mem_ptr def_mem(new mkldnn::memory(def_pd));