expression层中的output数据空间为空的bug

zjhellofss · Sep 1, 2023 · 132d598 · 132d598
1 parent bdf8d32
commit 132d598
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/source/layer/abstract/layer.cpp b/source/layer/abstract/layer.cpp
@@ -69,9 +69,6 @@ StatusCode Layer::Forward() {
 
   const std::shared_ptr<RuntimeOperand>& output_operand_datas =
       runtime_operator->output_operands;
-  if (output_operand_datas == nullptr || output_operand_datas->datas.empty()) {
-    int a = 3;
-  }
   CHECK(!layer_input_datas.empty())
       << runtime_operator->name << " Layer input data is empty";
   CHECK(output_operand_datas != nullptr && !output_operand_datas->datas.empty())

diff --git a/source/layer/details/expression.cpp b/source/layer/details/expression.cpp
@@ -115,8 +115,9 @@ StatusCode ExpressionLayer::Forward(
   std::vector<sftensor> output_node = op_stack.top();
   op_stack.pop();
   for (int i = 0; i < batch_size; ++i) {
-    CHECK(outputs.at(i) != nullptr && !outputs.at(i)->empty());
-    CHECK(outputs.at(i)->shapes() == output_node.at(i)->shapes());
+    if (outputs.at(i) != nullptr && !outputs.at(i)->empty()) {
+      CHECK(outputs.at(i)->shapes() == output_node.at(i)->shapes());
+    }
     outputs.at(i) = output_node.at(i);
   }
   return StatusCode::kSuccess;