Re: [PR] [DLight] Fix a corner case for reduction rule [tvm]

2024-04-05 Thread via GitHub
Hzfengsy merged PR #16848: URL: https://github.com/apache/tvm/pull/16848 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail:

[PR] [DLight] Fix a corner case for reduction rule [tvm]

2024-04-04 Thread via GitHub
Hzfengsy opened a new pull request, #16848: URL: https://github.com/apache/tvm/pull/16848 The current rule will fail when the output shape is only one element, because of missing `preserve_unit_loops`. This PR fixes it and adds a test case. -- This is an automated message from the Apache