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 MPS kernels #7643

Merged
merged 32 commits into from
Aug 1, 2023
Merged

Add MPS kernels #7643

merged 32 commits into from
Aug 1, 2023

Commits on Jun 20, 2023

  1. Draft

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    da9c2de View commit details
    Browse the repository at this point in the history

  2. NMS f32

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    7f0d4ce View commit details
    Browse the repository at this point in the history

  3. roi_align fw

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    c7c43dc View commit details
    Browse the repository at this point in the history

  4. roi_align bw (failed)

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    ccde29c View commit details
    Browse the repository at this point in the history

  5. roi_pool fw

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    c930e54 View commit details
    Browse the repository at this point in the history

  6. roi_pool bw (failed prec)

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    3305cc1 View commit details
    Browse the repository at this point in the history

  7. ps_roi_align fw

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    0f8d2c3 View commit details
    Browse the repository at this point in the history

  8. ps_roi_align bw (failed prec)

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    e157c7c View commit details
    Browse the repository at this point in the history

  9. Several improvements

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    160d5b5 View commit details
    Browse the repository at this point in the history

  10. ps_roi_pool fw

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    40ea525 View commit details
    Browse the repository at this point in the history

  11. ps_roi_pool bw

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    2c20036 View commit details
    Browse the repository at this point in the history

  12. Rename kernels header

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    a427c2a View commit details
    Browse the repository at this point in the history

  13. Add atol to RoI backward tests

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    8036dc2 View commit details
    Browse the repository at this point in the history

  14. mps kernels formatting

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    0ae9124 View commit details
    Browse the repository at this point in the history

  15. binaryPSO -> visionPSO

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    1d21cfc View commit details
    Browse the repository at this point in the history

  16. Formatting

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    3018b25 View commit details
    Browse the repository at this point in the history

  17. Testing

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    d609da4 View commit details
    Browse the repository at this point in the history

  18. Rename cpu_and_gpu to cpu_and_cuda

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    256bd56 View commit details
    Browse the repository at this point in the history

  19. formatting

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    990685f View commit details
    Browse the repository at this point in the history

  20. mps kernel dtype consistency

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    5dce2d7 View commit details
    Browse the repository at this point in the history

  21. Kernel improvements

    @qqaatw
    qqaatw committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    40ebde5 View commit details
    Browse the repository at this point in the history

Commits on Jun 21, 2023

  1. Merge branch 'main' of github.com:pytorch/vision into add_mps_kernels

    @NicolasHug
    NicolasHug committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    8e4d868 View commit details
    Browse the repository at this point in the history

Commits on Jul 3, 2023

  1. Merge branch 'main' into add_mps_kernels

    @NicolasHug
    NicolasHug authored Jul 3, 2023
    Configuration menu
    Copy the full SHA
    24109d4 View commit details
    Browse the repository at this point in the history

Commits on Jul 4, 2023

  1. Apply suggestions from code review

    @qqaatw
    qqaatw committed Jul 4, 2023
    Configuration menu
    Copy the full SHA
    efbb52e View commit details
    Browse the repository at this point in the history

  2. Test more dtypes for roi forward functions and assert half inputs in … 

    …MPS roi backward kernels
    @qqaatw
    qqaatw committed Jul 4, 2023
    Configuration menu
    Copy the full SHA
    b36cafa View commit details
    Browse the repository at this point in the history

Commits on Jul 7, 2023

  1. Add mps error inputs check

    @qqaatw
    qqaatw committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    fad54f6 View commit details
    Browse the repository at this point in the history

Commits on Jul 10, 2023

  1. Clean up headers

    @qqaatw
    qqaatw committed Jul 10, 2023
    Configuration menu
    Copy the full SHA
    66a00fc View commit details
    Browse the repository at this point in the history

Commits on Jul 17, 2023

  1. Fix dtype parameters

    @qqaatw
    qqaatw committed Jul 17, 2023
    Configuration menu
    Copy the full SHA
    70f3906 View commit details
    Browse the repository at this point in the history

  2. parameterize nms gpu test

    @qqaatw
    qqaatw committed Jul 17, 2023
    Configuration menu
    Copy the full SHA
    b1cf619 View commit details
    Browse the repository at this point in the history

Commits on Aug 1, 2023

  1. Merge branch 'main' into add_mps_kernels

    @NicolasHug
    NicolasHug authored Aug 1, 2023
    Configuration menu
    Copy the full SHA
    3f82ee4 View commit details
    Browse the repository at this point in the history

  2. Allow to skip MPS tests internally on non-MPS machines

    @NicolasHug
    NicolasHug committed Aug 1, 2023
    Configuration menu
    Copy the full SHA
    108bc15 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into add_mps_kernels

    @NicolasHug
    NicolasHug authored Aug 1, 2023
    Configuration menu
    Copy the full SHA
    c825c53 View commit details
    Browse the repository at this point in the history