复用runtime_op中的一部分内存

zjhellofss · Mar 17, 2024 · ab9d5a6 · ab9d5a6
1 parent ba2d096
commit ab9d5a6
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/include/runtime/runtime_op.hpp b/include/runtime/runtime_op.hpp
@@ -53,8 +53,6 @@ struct RuntimeOperatorBase {
 
   int32_t end_forward_index = -1;
 
-  int32_t swap_forward_index = -1;
-
   /// Whether this operator has run in current execution
   bool has_forward = false;
 

diff --git a/source/runtime/runtime_op.cpp b/source/runtime/runtime_op.cpp
@@ -142,9 +142,6 @@ void RuntimeOperatorUtils<float>::InitOperatorOutput(
         }
 
         const auto& prev_runtime_op = operators.at(j);
-        if (runtime_op->forward_index > prev_runtime_op->swap_forward_index) {
-          prev_runtime_op->end_forward_index = prev_runtime_op->swap_forward_index;
-        }
         if (!prev_runtime_op->output_operands || prev_runtime_op->end_forward_index == -1) {
           continue;
         }
@@ -163,7 +160,6 @@ void RuntimeOperatorUtils<float>::InitOperatorOutput(
                   std::make_shared<ftensor>(output_tensor->raw_ptr(), output_tensor->shapes());
               CheckAndReshapeTensor(output_tensors->datas[b], operand_shapes);
             }
-            prev_runtime_op->swap_forward_index = runtime_op->end_forward_index;
             prev_runtime_op->end_forward_index = -1;
           }
         }