You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by "Pratheesh-04-MCW (via GitHub)" <gi...@apache.org> on 2025/04/09 11:05:21 UTC
[PR] [Relax][PyTorch] Add Stack Op Support for Exported Program [tvm]
Pratheesh-04-MCW opened a new pull request, #17819:
URL: https://github.com/apache/tvm/pull/17819
This PR refactors the Stack Operation to be implemented at the core level using decomposed operations. It allows Stack to be accessed directly via R.Stack, making it more efficient and easier to use in TVM models.
--
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] [Relax][PyTorch] Add Stack Op Support for Exported Program [tvm]
Posted by "Hzfengsy (via GitHub)" <gi...@apache.org>.
Hzfengsy merged PR #17819:
URL: https://github.com/apache/tvm/pull/17819
--
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] [Relax][PyTorch] Add Stack Op Support for Exported Program [tvm]
Posted by "Pratheesh-04-MCW (via GitHub)" <gi...@apache.org>.
Pratheesh-04-MCW commented on PR #17819:
URL: https://github.com/apache/tvm/pull/17819#issuecomment-2812497287
@Hzfengsy Could you please review this PR?
--
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] [Relax][PyTorch] Add Stack Op Support for Exported Program [tvm]
Posted by "yongwww (via GitHub)" <gi...@apache.org>.
yongwww commented on code in PR #17819:
URL: https://github.com/apache/tvm/pull/17819#discussion_r2049330382
##########
python/tvm/relax/op/op_attrs.py:
##########
@@ -139,6 +139,11 @@ class SqueezeAttrs(Attrs):
"""Attributes for squeeze operator"""
+@tvm._ffi.register_object("relax.attrs.StackAttrs")
+class StackAttrs(Attrs):
+ """Attributes for concat operator"""
Review Comment:
typo: concat to stack
--
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