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

Adding GPU acceleration to encode_jpeg #8391

Merged
merged 21 commits into from
Jun 13, 2024
Merged

Adding GPU acceleration to encode_jpeg #8391

merged 21 commits into from
Jun 13, 2024

Commits on Apr 23, 2024

  1. Adding GPU acceleration to encode_jpeg 

    Summary:
I'm adding GPU support to the existing torchvision.io.encode_jpeg function. If the input tensors are on the GPU, the CUDA version will be used and the CPU version otherwise.
Additionally, I'm adding a new function torchvision.io.encode_jpegs (plural) with uses a fused kernel and may be faster than successive calls to the singular version which incurs kernel launch overhead for each call.
If it's alright, I'll be happy to refactor decode_jpeg to follow this
convention in a follow up PR.

Test Plan:
1. pytest test -vvv
2. ufmt format torchvision
3. flake8 torchvision

Reviewers:

Subscribers:

Tasks:

Tags:
    @deekay42
    deekay42 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    133d7c1 View commit details
    Browse the repository at this point in the history

  2. fix test cases

    @deekay42
    deekay42 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    4cc30cb View commit details
    Browse the repository at this point in the history

  3. fix lints

    @deekay42
    deekay42 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    2db02f0 View commit details
    Browse the repository at this point in the history

  4. fix lints2

    @deekay42
    deekay42 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    6acef83 View commit details
    Browse the repository at this point in the history

Commits on Apr 29, 2024

  1. latest round of updates

    @deekay42
    deekay42 committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    ae0450d View commit details
    Browse the repository at this point in the history

Commits on Apr 30, 2024

  1. fix lints

    @deekay42
    deekay42 committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    a799c53 View commit details
    Browse the repository at this point in the history

Commits on May 31, 2024

  1. Ignore mypy

    @NicolasHug
    NicolasHug committed May 31, 2024
    Configuration menu
    Copy the full SHA
    c5810ff View commit details
    Browse the repository at this point in the history

  2. Add comment

    @NicolasHug
    NicolasHug committed May 31, 2024
    Configuration menu
    Copy the full SHA
    ff40253 View commit details
    Browse the repository at this point in the history

  3. minor test refactor

    @NicolasHug
    NicolasHug committed May 31, 2024
    Configuration menu
    Copy the full SHA
    0972863 View commit details
    Browse the repository at this point in the history

  4. Merge branch 'main' of github.com:pytorch/vision into add_gpu_encode

    @NicolasHug
    NicolasHug committed May 31, 2024
    Configuration menu
    Copy the full SHA
    4ce658d View commit details
    Browse the repository at this point in the history

Commits on Jun 4, 2024

  1. Merge branch 'pytorch:main' into add_gpu_encode

    @deekay42
    deekay42 committed Jun 4, 2024
    Configuration menu
    Copy the full SHA
    65372a3 View commit details
    Browse the repository at this point in the history

Commits on Jun 5, 2024

  1. Caching nvjpeg vars across calls

    @deekay42
    deekay42 committed Jun 5, 2024
    Configuration menu
    Copy the full SHA
    62e072a View commit details
    Browse the repository at this point in the history

  2. Update if nvjpeg not found

    @deekay42
    deekay42 committed Jun 5, 2024
    Configuration menu
    Copy the full SHA
    f190d99 View commit details
    Browse the repository at this point in the history

Commits on Jun 10, 2024

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

    @NicolasHug
    NicolasHug committed Jun 10, 2024
    Configuration menu
    Copy the full SHA
    b5eaa89 View commit details
    Browse the repository at this point in the history

  2. Revert "Ignore mypy" 

    This reverts commit c5810ff.
    @NicolasHug
    NicolasHug committed Jun 10, 2024
    Configuration menu
    Copy the full SHA
    5051050 View commit details
    Browse the repository at this point in the history

  3. Add comment

    @NicolasHug
    NicolasHug committed Jun 10, 2024
    Configuration menu
    Copy the full SHA
    136f790 View commit details
    Browse the repository at this point in the history

Commits on Jun 11, 2024

  1. minor changes to address ahmad's comments

    @deekay42
    deekay42 committed Jun 11, 2024
    Configuration menu
    Copy the full SHA
    0a88d27 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'add_gpu_encode' of https://github.com/deekay42/vision i… 

    …nto add_gpu_encode
    @deekay42
    deekay42 committed Jun 11, 2024
    Configuration menu
    Copy the full SHA
    df60183 View commit details
    Browse the repository at this point in the history

Commits on Jun 12, 2024

  1. add dtor log messages

    @deekay42
    deekay42 committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    f3c8a72 View commit details
    Browse the repository at this point in the history

  2. Skip CUDA cleanup altogether

    @deekay42
    deekay42 committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    117d1f1 View commit details
    Browse the repository at this point in the history

Commits on Jun 13, 2024

  1. Merge branch 'main' into add_gpu_encode

    @NicolasHug
    NicolasHug committed Jun 13, 2024
    Configuration menu
    Copy the full SHA
    21eca4c View commit details
    Browse the repository at this point in the history