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.

Sign up for GitHub

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

Jump to bottom

cuda_torchdiffeq_integrator.py: check if pytorch is built with cuda capabilities #2613

Open

mitchdz wants to merge 1 commit into NVIDIA:main from mitchdz:mitch/add_pytorch_check

Collaborator

mitchdz commented Feb 13, 2025

Check if the installed pytorch is cuda aware.

This seems to only be an issue in the field on aarch64 systems where the pytorch in the default pip repository is not cuda aware.

This is a code check of the warning described in https://nvidia.github.io/cuda-quantum/latest/using/backends/dynamics.html#numerical-integrators.


          cuda_torchdiffeq_integrator.py: check if pytorch is built with cuda c…

7abf58c

…apabilities

Signed-off-by: Mitchell Dzurick <[email protected]>

copy-pr-bot bot commented Feb 13, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

schweitzpgi added the build label

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

build