Reorder2Default: return directly for default format (apache#10810)

rahul003 · May 4, 2018 · 5328f36 · 5328f36
1 parent 2b101b4
commit 5328f36
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));