Skip to content

[TIR][FIX] update FlopEstimator to include missing nodes#17598

Merged
wrongtest-intellif merged 6 commits into
apache:mainfrom
PatrikPerssonInceptron:feature/tuning-fix
Jan 28, 2025
Merged

[TIR][FIX] update FlopEstimator to include missing nodes#17598
wrongtest-intellif merged 6 commits into
apache:mainfrom
PatrikPerssonInceptron:feature/tuning-fix

Conversation

@PatrikPerssonInceptron

Copy link
Copy Markdown
Contributor

Updated FlopEstimator to include missing nodes that would otherwise cause tuning to fail for certain TIR functions

Comment thread src/tir/analysis/estimate_flops.cc Outdated
@wrongtest-intellif

Copy link
Copy Markdown
Contributor

@tvm-bot re-run

@wrongtest-intellif wrongtest-intellif merged commit 85556df into apache:main Jan 28, 2025
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
* updated estimate flops to include the AllocateNode

* added AttrStmtNode

* added a visit to the AttrStmtNode body

* added a visit to value of AttrStmtNode
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants