From c475da7820a42ee7925069b5e1ffb903918c0913 Mon Sep 17 00:00:00 2001
From: Jinhui Yuan <yuan.ms2@gmail.com>
Date: Wed, 8 Aug 2018 11:47:04 +0800
Subject: [PATCH] fix packed blob shape check (#1095)

---
 oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp | 1 -
 1 file changed, 1 deletion(-)

diff --git a/oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp b/oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp
index 1589b8256e..a92bd28bff 100644
--- a/oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp
+++ b/oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp
@@ -6,7 +6,6 @@ void MdDiffAccCompTaskNode::FixPackedBlobDescOfProducedRegst() {
   std::shared_ptr<RegstDesc> md_diff_acc_regst = GetProducedRegst("acc");
   CHECK(md_diff_acc_regst->IsLocked());
   Shape& shape = md_diff_acc_regst->MutBlobDesc(GenPackedLbi())->mut_shape();
-  CHECK_EQ(1, shape.NumAxes());
   shape = Shape({RoundUp(shape.elem_cnt(), parallel_ctx()->parallel_num())});
 }
 
-- 
GitLab