Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

[API] Add new dlpack API #20546

Merged
szha merged 19 commits into from
Nov 29, 2021
Merged

[API] Add new dlpack API #20546

szha merged 19 commits into from
Nov 29, 2021

Conversation

@barry-jin
Copy link
Contributor

@barry-jin barry-jin commented Aug 20, 2021

Description

Add new dlpack API specified in Array API standardizaiton #20501

References:

  1. Discussion on synchronization Specify synchronization semantics dmlc/dlpack#57
  2. New dlpack API in TVM [DLPACK] Support the new python array api with DLPack tvm#7993
  3. New dlpack API in pytorch Adds DLPack support pytorch/pytorch#57110

Checklist

Essentials

  • PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented

Changes

  • __dlpack__(self, stream=None) and __dlpack_device__(self)
  • Add synchronization between consumer and producer

Comments

  • If this change is a backward incompatible change, why must this change be made.
  • Interesting edge cases to note here

@mxnet-bot
Copy link

mxnet-bot commented Aug 20, 2021

Hey @barry-jin , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

  • To trigger all jobs: @mxnet-bot run ci [all]
  • To trigger specific jobs: @mxnet-bot run ci [job1, job2]

CI supported jobs: [edge, website, clang, unix-cpu, unix-gpu, sanity, centos-gpu, windows-cpu, centos-cpu, miscellaneous, windows-gpu]

Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.

@mseth10 mseth10 added the pr-work-in-progress PR is still work in progress label Aug 20, 2021
@barry-jin barry-jin changed the title [API] Add new dlpack API [WIP][API] Add new dlpack API Oct 26, 2021
@barry-jin
Copy link
Contributor Author

barry-jin commented Nov 2, 2021

@mxnet-bot run ci [unix-gpu]

@mxnet-bot
Copy link

mxnet-bot commented Nov 2, 2021

Jenkins CI successfully triggered : [unix-gpu]

@barry-jin
Copy link
Contributor Author

barry-jin commented Nov 7, 2021

@mxnet-bot run ci [unix-gpu]

@mxnet-bot
Copy link

mxnet-bot commented Nov 7, 2021

Jenkins CI successfully triggered : [unix-gpu]

@barry-jin barry-jin marked this pull request as ready for review November 8, 2021 17:59
@barry-jin barry-jin changed the title [WIP][API] Add new dlpack API [API] Add new dlpack API Nov 8, 2021
@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test and removed pr-work-in-progress PR is still work in progress pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 8, 2021
@barry-jin
Copy link
Contributor Author

barry-jin commented Nov 16, 2021

@mxnet-bot run ci [unix-cpu]

@mxnet-bot
Copy link

mxnet-bot commented Nov 16, 2021

Jenkins CI successfully triggered : [unix-cpu]

@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test pr-work-in-progress PR is still work in progress and removed pr-work-in-progress PR is still work in progress pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 16, 2021
@barry-jin
Copy link
Contributor Author

barry-jin commented Nov 16, 2021

@mxnet-bot run ci [unix-cpu]

@mxnet-bot
Copy link

mxnet-bot commented Nov 16, 2021

Jenkins CI successfully triggered : [unix-cpu]

@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test pr-awaiting-review PR is waiting for code review and removed pr-work-in-progress PR is still work in progress pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 16, 2021
@szha
Copy link
Member

szha commented Nov 23, 2021

@barry-jin the PR looks OK. Could you resolve the conflicts?

@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test pr-work-in-progress PR is still work in progress and removed pr-awaiting-review PR is waiting for code review pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 23, 2021
@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test pr-work-in-progress PR is still work in progress and removed pr-work-in-progress PR is still work in progress pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 23, 2021
@barry-jin
Copy link
Contributor Author

barry-jin commented Nov 23, 2021

@mxnet-bot run ci [miscellaneous]

@mxnet-bot
Copy link

mxnet-bot commented Nov 23, 2021

Jenkins CI successfully triggered : [miscellaneous]

@mseth10 mseth10 added pr-awaiting-testing PR is reviewed and waiting CI build and test pr-awaiting-review PR is waiting for code review and removed pr-work-in-progress PR is still work in progress pr-awaiting-testing PR is reviewed and waiting CI build and test labels Nov 23, 2021
@szha szha merged commit 45c7999 into apache:master Nov 29, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Reviewers

@aaronmarkham aaronmarkham Awaiting requested review from aaronmarkham aaronmarkham is a code owner

@eric-haibin-lin eric-haibin-lin Awaiting requested review from eric-haibin-lin eric-haibin-lin is a code owner

@marcoabreu marcoabreu Awaiting requested review from marcoabreu marcoabreu is a code owner

1 more reviewer

@szha szha szha left review comments

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

pr-awaiting-review PR is waiting for code review

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@barry-jin @mxnet-bot @szha @mseth10