Skip to content

Conversation

NicolasHug
Copy link
Contributor

@NicolasHug NicolasHug commented Oct 3, 2025

This PR lets our existing test suite check the BETA CUDA interface. The good news is that we pass the vast majority of the tests. The two failures, which are currently skipped with P0 TODOs, are related to end-of-stream logic. I'll follow-up with that in another PR. EDIT: I ended up fixing them here. We now pass all the tests!

This PR is based on #920, so to review it independently please look at f8f0402 (#921)

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Oct 3, 2025
@NicolasHug NicolasHug changed the title BETA CUDA interface: integrate with our existing tests BETA CUDA interface: integrate with our existing tests and fix EOF hang Oct 3, 2025
#
# TODONVDEC P2: Find a less clunky way to test the BETA CUDA interface. It
# will ultimately depend on how we want to publicly expose it.
if device_str == "cuda:0:beta":
Copy link
Contributor

Choose a reason for hiding this comment

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

Does this mean that our testing will only work on CUDA device 0?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

kinda. It's more that this line makes the assumption that we only ever test the beta interface as cuda:0:beta and never as cuda:1:beta.

We've never tested anything on cuda:1 so far, even for the default interface, but if we ever do this will be a trivial fix.

@NicolasHug NicolasHug merged commit 1177bb4 into meta-pytorch:main Oct 4, 2025
50 checks passed
@NicolasHug NicolasHug deleted the nvdec-tests branch October 4, 2025 10:15
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