-
Notifications
You must be signed in to change notification settings - Fork 182
[feat] GPU support #309
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
[feat] GPU support #309
Conversation
Dual is almost done, please check. Please consider applying for a GPU instance for testing cuda codes from GitHub Resources. |
real is supported |
For me, I only need forward. In addition, reverse is not easy to implement because it uses make_shared. |
How to add cuda-toolkit to the test environment? Usually, I use |
Hi @JieRen98 , thanks a lot for your contribution. Are you planning on more additions to this PR? |
For CUDA functions, I think they are mostly done. Of course, we can discuss.
Allan Leal ***@***.***>于2024年1月8日 周一12:53写道:
… Hi @JieRen98 <https://github.com/JieRen98> , thanks a lot for your
contribution. Are you planning on more additions to this PR?
—
Reply to this email directly, view it on GitHub
<#309 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/ALYD4VRPLOVOXAHD5M6IXYDYNO6ZRAVCNFSM6AAAAABBQMVKDCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTQOBQGY4DAMJQGQ>
.
You are receiving this because you were mentioned.Message ID:
***@***.***>
|
CUDA compile options |
Workflows may need to change too. |
This also removes the hard dependency on Eigen introduced in PR autodiff#309, as long as the user defines AUTODIFF_DEVICE_FUNC before including any autodiff headers.
This also removes the hard dependency on Eigen introduced in PR #309, as long as the user defines AUTODIFF_DEVICE_FUNC before including any autodiff headers.
No description provided.