Skip to content

Conversation

dvrogozh
Copy link
Contributor

This commit renames a cpu_and_cuda() function used for pytest device parameterization to all_supported_devices(). This is helpful for those who work on multi-gpu support in torchcodec as cpu_and_cuda() was one of the function required to be patched to allow running torchcodec tests on other devices. Rename steps out from binding function to cuda and allows to reduce the change scope to enable non-cuda devices.

CC: @scotts, @NicolasHug, @eromomon

This commit renames a `cpu_and_cuda()` function used for pytest
device parameterization to `all_supported_devices()`. This is
helpful for those who work on multi-gpu support in torchcodec as
`cpu_and_cuda()` was one of the function required to be patched
to allow running torchcodec tests on other devices. Rename steps
out from binding function to cuda and allows to reduce the change
scope to enable non-cuda devices.

Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>
@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Aug 14, 2025
Copy link
Contributor

@NicolasHug NicolasHug left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This sounds reasonable, thanks @dvrogozh

@NicolasHug NicolasHug merged commit 97bf7cc into meta-pytorch:main Aug 15, 2025
47 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants