[Paddle Inference] modify a check statement in memory_optimize_pass.cc (

PaddlePaddle#59638) [Paddle Inference] modify a check statement in memory_optimize_pass.cc
gouzil · Dec 5, 2023 · 0918f48 · 0918f48
1 parent 4059ec5
commit 0918f48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc
@@ -92,7 +92,7 @@ void MemoryOptimizePass::CollectLifeCycle(
 
           auto in_shape = node->Var()->GetShape();
           for (auto i : in_shape) {
-            CHECK_GT(i, 0);
+            CHECK_GE(i, 0);
           }
           auto var_bytes = std::accumulate(in_shape.begin(),
                                            in_shape.end(),