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 GPU->CPU transfers #5593

Merged
merged 9 commits into from
Sep 11, 2024
Merged

Enable GPU->CPU transfers #5593

merged 9 commits into from
Sep 11, 2024

Commits on Sep 10, 2024

  1. .cpu() works with experimental executor. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    494a626 View commit details
    Browse the repository at this point in the history

  2. Simplify copy. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    831cc9f View commit details
    Browse the repository at this point in the history

  3. Fix a bug in DataNode.cpu() with conditionals. 

    Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    3ff9ea9 View commit details
    Browse the repository at this point in the history

  4. Add dynamic_execution parameter to the Pipeline constructor. 

    Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    c15c009 View commit details
    Browse the repository at this point in the history

  5. Add C API support. 

    Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    1826e37 View commit details
    Browse the repository at this point in the history

  6. Unblock CPU operators taking GPU inputs. Adjust tests. Add test with … 

    …CPU `shapes` taking GPU input.

Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    5fccaa4 View commit details
    Browse the repository at this point in the history

  7. Fix misplaced comment. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    13f6e17 View commit details
    Browse the repository at this point in the history

  8. Restore error when specifying GPU argument inputs. Fix shapes_gpu test. 

    Signed-off-by: Michał Zientkiewicz <mzient@gmail.com>
    @mzient
    mzient committed Sep 10, 2024
    Configuration menu
    Copy the full SHA
    9f239a4 View commit details
    Browse the repository at this point in the history

Commits on Sep 11, 2024

  1. Refactor DataNode.cpu/gpu. Add backend tests in pipeline_test.py. 

    Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
    @mzient
    mzient committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    49e85e7 View commit details
    Browse the repository at this point in the history