Skip to content

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