You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/05/06 18:53:32 UTC

[GitHub] [tvm] YuchenJin opened a new pull request #7993: [DLPACK] Support the new python array api with DLPack

YuchenJin opened a new pull request #7993:
URL: https://github.com/apache/tvm/pull/7993


   Adopt the new [Python array API standard with DLPack](https://data-apis.org/array-api/latest/design_topics/data_interchange.html#syntax-for-data-interchange-with-dlpack):
   
   - Add `__dlpack__(self, stream=None)` and `__dlpack_device__(self)` methods on `tvm.nd.NDArray`. Note that tvm operates on the legacy default stream by default, so no explicit synchronization is needed.
   - `tvm.nd.from_dlpack` can take a tensor-like object with `__dlpack__` method, or a DLPack capsule.
   
   
   More contexts:
   - API design discussion: https://github.com/dmlc/dlpack/issues/57
   - PyTorch support: https://github.com/pytorch/pytorch/issues/55090, https://github.com/pytorch/pytorch/pull/57110


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] YuchenJin commented on pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
YuchenJin commented on pull request #7993:
URL: https://github.com/apache/tvm/pull/7993#issuecomment-833891650


   > Thanks Yuchen for the contribution! Overall it looks good to me. Please fix the lint issues
   
   


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] junrushao1994 merged pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
junrushao1994 merged pull request #7993:
URL: https://github.com/apache/tvm/pull/7993


   


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] junrushao1994 commented on pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
junrushao1994 commented on pull request #7993:
URL: https://github.com/apache/tvm/pull/7993#issuecomment-834086813


   Thanks @YuchenJin! It is now merged!


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] YuchenJin commented on pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
YuchenJin commented on pull request #7993:
URL: https://github.com/apache/tvm/pull/7993#issuecomment-833779402


   cc @tqchen @junrushao1994 


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] YuchenJin edited a comment on pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
YuchenJin edited a comment on pull request #7993:
URL: https://github.com/apache/tvm/pull/7993#issuecomment-833891650


   Thanks, @junrushao1994! :)


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] YuchenJin closed pull request #7993: [DLPACK] Support the new python array api with DLPack

Posted by GitBox <gi...@apache.org>.
YuchenJin closed pull request #7993:
URL: https://github.com/apache/tvm/pull/7993


   


-- 
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org