Skip to content

Don't fuse take with dynamic inputs - #6979

Merged
masahi merged 4 commits into
apache:mainfrom
mbrookhart:dynamic_take
Nov 26, 2020
Merged

Don't fuse take with dynamic inputs#6979
masahi merged 4 commits into
apache:mainfrom
mbrookhart:dynamic_take

Conversation

@mbrookhart

Copy link
Copy Markdown
Contributor

Follow up to #6906

This stops fusion on take if it's inputs are dynamic and adds a regression test.

Comment threadpython/tvm/relay/op/_transform.py Outdated
Comment threadpython/tvm/topi/transform.py Outdated
@masahi
masahi merged commit f974166 into apache:mainNov 26, 2020
@masahi

Copy link
Copy Markdown
Member

thanks @mbrookhart

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* add a regression test for fusing dynamic take
* add legalize for take that stops fusion on dynamic inputs
* fix lint
* fix typo
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* add a regression test for fusing dynamic take
* add legalize for take that stops fusion on dynamic inputs
* fix lint
* fix typo
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* add a regression test for fusing dynamic take
* add legalize for take that stops fusion on dynamic inputs
* fix lint
* fix typo
Sign up for freeto 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

@mbrookhart@masahi