修改内存复用时的条件判断

zjhellofss · Mar 20, 2024 · 8506233 · 8506233
1 parent 8bfadb9
commit 8506233
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/runtime/runtime_op.cpp b/source/runtime/runtime_op.cpp
@@ -142,7 +142,7 @@ void RuntimeOperatorUtils<float>::InitOperatorOutput(
         }
 
         if (runtime_op->start_time > prev_runtime_op->end_time) {
-          if (prev_runtime_op->output_operands->size() >= operand_size) {
+          if (prev_runtime_op->output_operands->size() == operand_size) {
             has_found = true;
             const auto& prev_output_operand = prev_runtime_op->output_operands;
             runtime_op->output_operands = std::make_shared<RuntimeOperand>();