This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 561d0859740456c84b45f0a548ed9df9dbe62d9d Author: Qingchao Shen <[email protected]> AuthorDate: Tue May 13 13:26:07 2025 +0800 Update estimate_flops.cc --- src/tir/analysis/estimate_flops.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/tir/analysis/estimate_flops.cc b/src/tir/analysis/estimate_flops.cc index 586ee04197..c8bea6ecd0 100644 --- a/src/tir/analysis/estimate_flops.cc +++ b/src/tir/analysis/estimate_flops.cc @@ -174,10 +174,10 @@ class FlopEstimator : private ExprFunctor<TResult(const PrimExpr& n)>, } TResult VisitStmt_(const WhileNode* op) override { - // TODO(jikechao): Improve while loop FLOP estimation with loop bound analysis - TResult result = VisitExpr(op->condition); - result += VisitStmt(op->body); - return result; + // TODO(jikechao): Improve while loop FLOP estimation with loop bound analysis + TResult result = VisitExpr(op->condition); + result += VisitStmt(op->body); + return result; } TResult VisitStmt_(const LetStmtNode* let) override {
