-
Notifications
You must be signed in to change notification settings - Fork 354
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
MPS with cuQuantum #2168
base: main
Are you sure you want to change the base?
MPS with cuQuantum #2168
Conversation
Because |
It fails running with MPS method on GPU with error message as following, |
Yes, I know I am actively working on this please give me some time and I
will let you know when I am done
…On Fri, 14 Jun, 2024, 7:22 am Jun Doi, ***@***.***> wrote:
It fails running with MPS method on GPU with error message as following,
ERROR TensorNet::contractor : CUTENSORNET_STATUS_INVALID_VALUE
—
Reply to this email directly, view it on GitHub
<#2168 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/ANBTUULPA33H5DXTGKNAQ4LZHJEGFAVCNFSM6AAAAABI4MTL7CVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDCNRXGA3DQMBQGI>
.
You are receiving this because you authored the thread.Message ID:
***@***.***>
|
@doichanj , please see if this is good enough :) and, extremely sorry, for the delay, doing any development in cloud VM, is not a good experience. |
I have absolutely no idea why macOS tests are failing. |
Looks like the problem is old cvxpy version, #2169 should fix it? |
I have got my hands on Nvidia 3060 12 GB. |
I was using cuda 12.1 |
Summary
This PR aims to add a feature of doing matrix-product-state simulation on Nvidia GPUs with cutensor of cuQuantum.
Looks like its working, and the results are correct too :)
Details and comments
Shows performance gains,
I am not super happy with the performance gains, but I think time is running out now, and I have to leave this PR here, but, I will surely get back here and do a follow up PR which will show improvements.
fixes #2112