Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add dlpack support #7025

Merged
merged 19 commits into from
May 22, 2024
Merged

Add dlpack support #7025

merged 19 commits into from
May 22, 2024

Commits on May 21, 2024

  1. added the test and the pybinding.

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    822d169 View commit details
    Browse the repository at this point in the history

  2. Add toDLPack and it compiles.

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    70996c4 View commit details
    Browse the repository at this point in the history

  3. finished fromDLPack and it compiles

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    09ff265 View commit details
    Browse the repository at this point in the history

  4. fixed an unrecognized symbol issue.

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    1f0ed12 View commit details
    Browse the repository at this point in the history

  5. Change test to use .to(xla_device).

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    b3516c2 View commit details
    Browse the repository at this point in the history

  6. check pjrt_buffer is not null. Also in toDlPack, avoid doing from dev… 

    …ice to host transfer.
    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    f78d320 View commit details
    Browse the repository at this point in the history

  7. Improved the basic test

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    3e2ff67 View commit details
    Browse the repository at this point in the history

  8. add release GIL in to_dlpack.

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    01691c0 View commit details
    Browse the repository at this point in the history

  9. fixed a bug of not passing in the pjrt_buffer

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    251ae89 View commit details
    Browse the repository at this point in the history

  10. in the middle of fixing holds_[i] == 0 (1 vs. 0)

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    aac102b View commit details
    Browse the repository at this point in the history

  11. fixed the issue: pjrt_stream_executor_client.cc:1322] Check failed: h… 

    …olds_[i] == 0 (1 vs. 0)
    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    57a936e View commit details
    Browse the repository at this point in the history

  12. improve the unit tests. All test passed.

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    9ff3dd9 View commit details
    Browse the repository at this point in the history

  13. Clean up unused prints and comments

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    8dd8381 View commit details
    Browse the repository at this point in the history

  14. linter

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    60b4d59 View commit details
    Browse the repository at this point in the history

  15. fix comment

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    93dba65 View commit details
    Browse the repository at this point in the history

  16. fix comments

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    867dd53 View commit details
    Browse the repository at this point in the history

  17. fix comment

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    345945e View commit details
    Browse the repository at this point in the history

  18. linter

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    2e8ff4a View commit details
    Browse the repository at this point in the history

  19. fix comments and run linter

    @vanbasten23
    vanbasten23 committed May 21, 2024
    Configuration menu
    Copy the full SHA
    98cc739 View commit details
    Browse the repository at this point in the history