-
Notifications
You must be signed in to change notification settings - Fork 149
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
[WIP] Include GPU in wheels #479
base: master
Are you sure you want to change the base?
Conversation
Bazel page is returning 404s, need to wait for it to come back and retry. |
96f86eb
to
74942aa
Compare
2c6d87d
to
29d4993
Compare
An inevitable side effect of building GPU support on machines that lack GPU support: we have no way to test GPU behavior in continuous integration. However, since the wheels generated by GHA are accessible from the test artifacts, it will be possible for us to spin up a GPU-enabled machine, copy over the wheels, and test before release. If we move forward with this, the process outlined above will become a mandatory part of all GPU changes. |
@95-martin-orion @sergeisakov FYI, we prepared a demo to show how to build a project upon CUDA/cuQuantum wheels: https://github.com/NVIDIA/cuQuantum/tree/main/extra/demo_build_with_wheels |
Fixes #430.
Since cross-platform support requires building on the manylinux image, installing NVCC on the runner is the best we can do unless we decide to push a manylinux + CUDA image to Docker Hub.