You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by "eaten-cake (via GitHub)" <gi...@apache.org> on 2025/05/25 11:46:51 UTC
[PR] [TOPI][NN][Layer_Norm] Fix layer_norm error with reduce-only axes [tvm]
eaten-cake opened a new pull request, #18012:
URL: https://github.com/apache/tvm/pull/18012
Fixes #18002
When all dimensions of layer_norm input are reduction axes, the variable `non_reduce_indices` becomes empty, while the previous variable `target_shape` has `atleast1d` parameter enabled, causing shape inconsistency.
https://github.com/apache/tvm/blob/2d964b4133aac2f92e4185b3f095df4eb3bf3a90/include/tvm/topi/nn/layer_norm.h#L111-L121
https://github.com/apache/tvm/blob/2d964b4133aac2f92e4185b3f095df4eb3bf3a90/include/tvm/topi/nn/layer_norm.h#L67-L68
Also added tests for reduce-only axes cases
--
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: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
Re: [PR] [TOPI][NN][Layer_Norm] Fix layer_norm error with reduce-only axes [tvm]
Posted by "eaten-cake (via GitHub)" <gi...@apache.org>.
eaten-cake commented on PR #18012:
URL: https://github.com/apache/tvm/pull/18012#issuecomment-2927256183
cc @MasterJH5574 @Hzfengsy @tqchen @mshr-h
--
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: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
Re: [PR] [TOPI][NN][Layer_Norm] Fix layer_norm error with reduce-only axes [tvm]
Posted by "eaten-cake (via GitHub)" <gi...@apache.org>.
eaten-cake closed pull request #18012: [TOPI][NN][Layer_Norm] Fix layer_norm error with reduce-only axes
URL: https://github.com/apache/tvm/pull/18012
--
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: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
Re: [PR] [TOPI][NN][Layer_Norm] Fix layer_norm error with reduce-only axes [tvm]
Posted by "eaten-cake (via GitHub)" <gi...@apache.org>.
eaten-cake commented on PR #18012:
URL: https://github.com/apache/tvm/pull/18012#issuecomment-2909837287
cc @tqchen @Hzfengsy @mshr-h
--
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: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org