vision
vision copied to clipboard
Add gpu decode
Over 8000 imgs/s on 1 A100 GPU
Platform: Linux-5.12.0-0_fbk7_zion_6511_gd766966f605a-x86_64-with-glibc2.34
Logical CPUs: 192
CUDA device: NVIDIA PG509-210
Total Memory: 84.99 GB
Mean image size: 551x676
[---------------------------------------------------------------- Image Decoding ----------------------------------------------------------------]
| 1 images | 100 images | 1000 images
1 threads: ---------------------------------------------------------------------------------------------------------------------------------------
CPU (unfused): [torchvision.io.decode_jpeg(img, device='cpu') for img in encoded_images_trunc] | 3301.9 | 271141.6 | 2541465.3
CPU (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cpu') | 3239.7 | 288522.8 | 2596394.3
CUDA (unfused): [torchvision.io.decode_jpeg(img, device='cuda') for img in encoded_images_trunc] | 603.7 | 60097.8 | 573783.4
CUDA (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cuda') | 600.6 | 12972.6 | 127654.8
12 threads: --------------------------------------------------------------------------------------------------------------------------------------
CPU (unfused): [torchvision.io.decode_jpeg(img, device='cpu') for img in encoded_images_trunc] | 3330.5 | 272498.9 | 2552944.3
CPU (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cpu') | 3339.7 | 257796.7 | 2511005.4
CUDA (unfused): [torchvision.io.decode_jpeg(img, device='cuda') for img in encoded_images_trunc] | 603.8 | 59138.0 | 588341.4
CUDA (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cuda') | 605.0 | 13163.7 | 127891.4
24 threads: --------------------------------------------------------------------------------------------------------------------------------------
CPU (unfused): [torchvision.io.decode_jpeg(img, device='cpu') for img in encoded_images_trunc] | 3227.5 | 276357.8 | 2518914.3
CPU (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cpu') | 3277.7 | 257554.9 | 2497894.3
CUDA (unfused): [torchvision.io.decode_jpeg(img, device='cuda') for img in encoded_images_trunc] | 607.9 | 58306.1 | 583932.6
CUDA (fused): torchvision.io.decode_jpeg(encoded_images_trunc, device='cuda') | 653.2 | 12604.1 | 124130.5
Times are in microseconds (us).```
:link: Helpful Links
:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/8496
- :page_facing_up: Preview Python docs built from this PR
Note: Links to docs will display an error until the docs builds have been completed.
:x: 13 New Failures, 6 Unrelated Failures
As of commit efa746d8b4892925c65d370e89f9357fb267dac9 with merge base 5242d6ac27c5eae7b74b65f28bd1373de955327e ():
NEW FAILURES - The following jobs have failed:
- CMake / macos (macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - CMake / windows (windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128 - Lint / bc (gh)
Process completed with exit code 1. - Tests / unittests-macos (3.10, macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-macos (3.11, macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-macos (3.12, macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-macos (3.8, macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-macos (3.9, macos-m1-stable) / macos-job (gh)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-windows (3.10, windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128 - Tests / unittests-windows (3.11, windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128 - Tests / unittests-windows (3.12, windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128 - Tests / unittests-windows (3.8, windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128 - Tests / unittests-windows (3.9, windows.4xlarge, cpu) / windows-job (gh)
The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
FLAKY - The following jobs failed but were likely due to flakiness present on trunk:
- CMake / linux (linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-linux (3.10, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-linux (3.11, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-linux (3.12, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-linux (3.8, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128 - Tests / unittests-linux (3.9, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
The process '/usr/bin/git' failed with exit code 128
This comment was automatically generated by Dr. CI and updates every 15 minutes.
Hey @NicolasHug!
You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py