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

[Bugfix] Preserve IRModule type definition and imports in NameMangleExtFuncs #8523

Merged
merged 4 commits into from
Jul 29, 2021

Conversation

zxy844288792
Copy link
Contributor

@zxy844288792 zxy844288792 commented Jul 21, 2021

When calling graph_partition pass on tensoraay ops, it fails with errorCheck failed: (it != type_definitions.end()) is false: There is no definition of static_tensor_float32_any_2_3_t This is introduced by #8014.

This fix imports type definations from old module when craeting a new module in NameMangleExtFuncs.

cc @comaniac

tests/python/relay/test_tensor_array.py Outdated Show resolved Hide resolved
@comaniac comaniac changed the title [Bugfix] check type failed when partitioning tensorarray ops [Bugfix] Preserve IRModule type definition and imports in NameMangleExtFuncs Jul 21, 2021
@comaniac
Copy link
Contributor

cc @manupa-arm

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

LGTM

tests/python/relay/test_pass_partition_graph.py Outdated Show resolved Hide resolved
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
Copy link
Contributor

@manupak manupak left a comment

Choose a reason for hiding this comment

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

Thanks for the fix! LGTM!

@comaniac comaniac merged commit e39204e into apache:main Jul 29, 2021
@comaniac
Copy link
Contributor

Thanks @zxy844288792 @manupa-arm

ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
…xtFuncs (apache#8523)

* bug fix and add tensorarray with partition pass test case

* change test function location and address comments

* Update tests/python/relay/test_pass_partition_graph.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* trigger CI

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
…xtFuncs (apache#8523)

* bug fix and add tensorarray with partition pass test case

* change test function location and address comments

* Update tests/python/relay/test_pass_partition_graph.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* trigger CI

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
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