fix packed blob shape check (#1095)

c475da78 · Jinhui Yuan · GitHub · 18bda5ca · c475da78
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

oneflow/core/graph/model_diff_accumulate_compute_task_node.cpp ...ow/core/graph/model_diff_accumulate_compute_task_node.cpp +0 -1

未找到文件。
--- 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())});
 }