Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Simplify full broadcast #7423

Merged
merged 6 commits into from
Feb 10, 2021
Merged

Conversation

mbrookhart
Copy link
Contributor

This PR simplifies full like ops that can be fused into broadcast ops, such that we can enable constant folding these ops in other situations to fix various issues we've seen with dynamic models.

This also refactors the SimplifyExpr pass somewhat to ease the addition of future patterns.

@vinx13 @jwfromm @icemelon9

@vinx13 vinx13 merged commit 12c6b70 into apache:main Feb 10, 2021
@vinx13
Copy link
Member

vinx13 commented Feb 10, 2021

Thanks @mbrookhart @masahi

@mbrookhart mbrookhart deleted the simplify_full_broadcast branch February 10, 2021 20:22
alexwong pushed a commit to alexwong/tvm that referenced this pull request Feb 11, 2021
* convert argwhere(full(const)) to reshape(arange())

* Add IsWildcard syntatic sugar

* add a simplify expression to fold full into broadcast ops

* Allow constant folding of full-like ops after SimplifyExpr

* fix a bug with the Attr Pattern matching

* remove skip_list
Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
* convert argwhere(full(const)) to reshape(arange())

* Add IsWildcard syntatic sugar

* add a simplify expression to fold full into broadcast ops

* Allow constant folding of full-like ops after SimplifyExpr

* fix a bug with the Attr Pattern matching

* remove skip_list
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
* convert argwhere(full(const)) to reshape(arange())

* Add IsWildcard syntatic sugar

* add a simplify expression to fold full into broadcast ops

* Allow constant folding of full-like ops after SimplifyExpr

* fix a bug with the Attr Pattern matching

* remove skip_list
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants