(tvm) branch main updated: Fix FLOP estimation for EvaluateNode by implementing VisitStmt_ handler (#17974)

[tqchen]

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


The following commit(s) were added to refs/heads/main by this push:
     new bcae402e6c Fix FLOP estimation for EvaluateNode by implementing 
VisitStmt_ handler (#17974)
bcae402e6c is described below

commit bcae402e6c305c4e3cb07c7f09a89cc669ffd675
Author: Qingchao Shen <qingchaos...@outlook.com>
AuthorDate: Wed May 28 18:40:08 2025 +0800

    Fix FLOP estimation for EvaluateNode by implementing VisitStmt_ handler 
(#17974)
    
    Update estimate_flops.cc
---
 src/tir/analysis/estimate_flops.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/tir/analysis/estimate_flops.cc 
b/src/tir/analysis/estimate_flops.cc
index 688fcd2263..df41e7da18 100644
--- a/src/tir/analysis/estimate_flops.cc
+++ b/src/tir/analysis/estimate_flops.cc
@@ -200,6 +200,7 @@ class FlopEstimator : private ExprFunctor<TResult(const 
PrimExpr& n)>,
   TResult VisitStmt_(const AllocateConstNode* op) override { return 
VisitStmt(op->body); }
   TResult VisitStmt_(const AllocateNode* op) override { return 
VisitStmt(op->body); }
   TResult VisitStmt_(const DeclBufferNode* op) override { return 
VisitStmt(op->body); }
+  TResult VisitStmt_(const EvaluateNode* op) override { return TResult(); }
 
   TResult VisitStmt_(const SeqStmtNode* seq) override {
     TResult result;