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

[Relay][Pass]Improve memory_allocation pass to support multiple i/o dynamic kernels #4595

Merged
merged 8 commits into from
Jan 4, 2020

Conversation

kevinthesun
Copy link
Contributor

This PR enhance memory_allocation pass to support dynamic shape kernels such as split and concatenate.

@zhiics @jroesch @yongwww

@kevinthesun kevinthesun changed the title Improve memory_allocation pass to support multiple i/o dynamic kernels [Relay][Pass]Improve memory_allocation pass to support multiple i/o dynamic kernels Dec 28, 2019
Copy link
Member

@zhiics zhiics left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM in general. @jroesch can you take a look as well when you get a chance?

python/tvm/relay/memory_alloc.py Outdated Show resolved Hide resolved
Copy link
Member

@yongwww yongwww left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

python/tvm/relay/op/_tensor.py Show resolved Hide resolved
@zhiics zhiics merged commit 9c638f0 into apache:master Jan 4, 2020
@zhiics
Copy link
Member

zhiics commented Jan 4, 2020

Thanks @kevinthesun @yongwww

alexwong pushed a commit to alexwong/tvm that referenced this pull request Feb 26, 2020
…ynamic kernels (apache#4595)

* Add more shape funcs

* Fix test

* Enhance test_any_concat

* Fix pylint

* Minor fix test

* Fix pylint

* Minor refactor

* Add test any for elemwise
alexwong pushed a commit to alexwong/tvm that referenced this pull request Feb 28, 2020
…ynamic kernels (apache#4595)

* Add more shape funcs

* Fix test

* Enhance test_any_concat

* Fix pylint

* Minor fix test

* Fix pylint

* Minor refactor

* Add test any for elemwise
zhiics pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2020
…ynamic kernels (apache#4595)

* Add more shape funcs

* Fix test

* Enhance test_any_concat

* Fix pylint

* Minor fix test

* Fix pylint

* Minor refactor

* Add test any for elemwise
@kevinthesun kevinthesun deleted the AddShapeFunc branch March 23, 2020 17:33
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.

3 participants