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

Enable Pytorch to share same memory pool as RMM via cli #1392

Merged
merged 8 commits into from
Oct 12, 2024
Merged

Enable Pytorch to share same memory pool as RMM via cli #1392

merged 8 commits into from
Oct 12, 2024

Commits on Oct 8, 2024

  1. POC pytorch in dask-cuda 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    12e0051 View commit details
    Browse the repository at this point in the history

  2. Allow passing in a list 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    a2bf28f View commit details
    Browse the repository at this point in the history

  3. Fix cli problem 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    d046478 View commit details
    Browse the repository at this point in the history

  4. Address Peters review 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    35647ae View commit details
    Browse the repository at this point in the history

  5. Style fixes 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    494ada6 View commit details
    Browse the repository at this point in the history

  6. Add a check for string input to allow a single string 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 8, 2024
    Configuration menu
    Copy the full SHA
    94982f6 View commit details
    Browse the repository at this point in the history

Commits on Oct 9, 2024

  1. Address Peters Review

    @VibhuJawa
    VibhuJawa committed Oct 9, 2024
    Configuration menu
    Copy the full SHA
    33871ac View commit details
    Browse the repository at this point in the history

Commits on Oct 11, 2024

  1. Add support for comma separated strings 

    Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
    @VibhuJawa
    VibhuJawa committed Oct 11, 2024
    Configuration menu
    Copy the full SHA
    2517874 View commit details
    Browse the repository at this point in the history