PatrikPerssonInceptron commented on code in PR #17598:
URL: https://github.com/apache/tvm/pull/17598#discussion_r1923479734
##########
src/tir/analysis/estimate_flops.cc:
##########
@@ -138,6 +138,7 @@ class FlopEstimator : private ExprFunctor<TResult(const
PrimExpr& n)>,
}
TResult VisitExpr_(const BufferLoadNode* op) override { return TResult(); }
+ TResult VisitStmt_(const AttrStmtNode* op) override { return TResult(); }
Review Comment:
I was a bit unsure about this when I added the node, but you are right,
there could technically be valid mops in the body. I updated the visitor to
visit the body of AttrStmtNode. In addition I added a visit to the value as
well.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
