-
Notifications
You must be signed in to change notification settings - Fork 62
BETA CUDA interface: integrate with our existing tests and fix EOF hang #921
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.
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
Conversation
cc295eb
to
f8f0402
Compare
# | ||
# 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": |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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.
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)