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

Implement interface for data transfer across GPU devices. #2308

Merged
merged 29 commits into from
Aug 21, 2023
Merged

Implement interface for data transfer across GPU devices. #2308

merged 29 commits into from
Aug 21, 2023

Commits on Aug 20, 2023

  1. Adding new get_device method to return a CUDA device with particular 

    ordinal.
    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    3d5d849 View commit details
    Browse the repository at this point in the history

  2. Adding an adapt function for AbstractArray to handle movement across 

    devices.
    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    f720d5f View commit details
    Browse the repository at this point in the history

  3. Making the get_device interface simpler, and some minor changes.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    40e085c View commit details
    Browse the repository at this point in the history

  4. Adding CPU option to get_device.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    652bf95 View commit details
    Browse the repository at this point in the history

  5. Removing KernelAbstractions from deps.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    b3cd292 View commit details
    Browse the repository at this point in the history

  6. Adding new get_device method to return a particular AMD device.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    9925c5b View commit details
    Browse the repository at this point in the history

  7. Adding new adapt_storage function for moving arrays. Also passing 

    ordinal information through `FluxAMDAdaptor`.
    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    df577f3 View commit details
    Browse the repository at this point in the history

  8. Moving relevant function definitions to extensions.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    8aa7eed View commit details
    Browse the repository at this point in the history

  9. Making _metal accept an ordinal.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    f137080 View commit details
    Browse the repository at this point in the history

  10. Adding new get_device method to return particular Metal device.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    ef265eb View commit details
    Browse the repository at this point in the history

  11. Adding new adapt_storage method for metal arrays.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    3fbb4f5 View commit details
    Browse the repository at this point in the history

  12. Fixing minor error.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    2c6bc55 View commit details
    Browse the repository at this point in the history

  13. Fixing minor error and spelling mistake.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    829dcfa View commit details
    Browse the repository at this point in the history

  14. Fixing package name: AMDGPU instead of AMD.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    930d29c View commit details
    Browse the repository at this point in the history

  15. Reverting back to old metal functionality.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    fedee3b View commit details
    Browse the repository at this point in the history

  16. Adding tests for moving models between CPU and NVIDIA devices.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    a4449f8 View commit details
    Browse the repository at this point in the history

  17. Adding tests for data movement on AMD devices.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    2a14650 View commit details
    Browse the repository at this point in the history

  18. Fixing index error while choosing AMD gpu device.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    a9fb328 View commit details
    Browse the repository at this point in the history

  19. Fixing AMD ordinal starting index.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    91a756f View commit details
    Browse the repository at this point in the history

  20. Adding docstring for new get_device method.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    a9f6615 View commit details
    Browse the repository at this point in the history

  21. Removing global name conflicts in tests.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    b47a6f4 View commit details
    Browse the repository at this point in the history

  22. Minor fix to AMDs device id tests.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    f1ab569 View commit details
    Browse the repository at this point in the history

  23. Disambiguating test variables.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    129a0b5 View commit details
    Browse the repository at this point in the history

  24. Adding more info in docstring of get_device, and writing some 

    documentation in the guide.
    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    7a5b847 View commit details
    Browse the repository at this point in the history

  25. Fixing minor error in AMD code.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    1b770bc View commit details
    Browse the repository at this point in the history

  26. Fixing yet another ordinal index error in AMD code.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    8c09485 View commit details
    Browse the repository at this point in the history

  27. Fixing another ordinal index error in AMD code.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    e367598 View commit details
    Browse the repository at this point in the history

  28. Fixing spelling mistake.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    08b1962 View commit details
    Browse the repository at this point in the history

  29. Replacing type checks for nothing but equality checks.

    @codetalker7
    codetalker7 committed Aug 20, 2023
    Configuration menu
    Copy the full SHA
    0063bc0 View commit details
    Browse the repository at this point in the history