Skip to content
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

Draft
wants to merge 43 commits into
base: master
Choose a base branch
from
Draft

[WIP] Include GPU in wheels #479

wants to merge 43 commits into from

Conversation

95-martin-orion
Copy link
Collaborator

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.

@google-cla google-cla bot added the cla: yes Override CLA status to unblock PR. label Nov 16, 2021
@95-martin-orion
Copy link
Collaborator Author

Bazel page is returning 404s, need to wait for it to come back and retry.

@95-martin-orion
Copy link
Collaborator Author

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.

@leofang
Copy link

leofang commented Jan 26, 2023

@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
Let us know if it helps!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes Override CLA status to unblock PR.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Provide GPU support in released qsimcirq wheel
3 participants