Skip to content

[Relay] Disable exception for ADT in mixed precision pass#15533

Merged
echuraev merged 1 commit intoapache:mainfrom
echuraev:echuraev/fix_to_mixed_precision_for_while_loop
Aug 14, 2023
Merged

[Relay] Disable exception for ADT in mixed precision pass#15533
echuraev merged 1 commit intoapache:mainfrom
echuraev:echuraev/fix_to_mixed_precision_for_while_loop

Conversation

@echuraev
Copy link
Copy Markdown
Contributor

If topology contains while loop and we want to transform it to mixed precision then we get an exception that "ADT are not supported for mixed precision pass". It happens, because while loop implemented as a lambda which is assigned to a VarNode.

In this commit I changed the behavior of ToMixedPrecision pass and instead of generating exception, it just do nothing.

Correspondent regression test is added.

cc: @AndrewZhaoLuo

If topology contains while loop and we want to transform it to mixed
precision then we get an exception that "ADT are not supported for
mixed precision pass". It happens, because while loop implemented as a
lambda which is assigned to a VarNode.

In this commit I changed the behavior of ToMixedPrecision pass and
instead of generating exception, it just do nothing.

Correspondent regression test is added.
@echuraev echuraev requested a review from AndrewZhaoLuo August 11, 2023 20:39
@tvm-bot
Copy link
Copy Markdown
Collaborator

tvm-bot commented Aug 11, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@echuraev echuraev merged commit 927df59 into apache:main Aug 14, 2023
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