-
-
Notifications
You must be signed in to change notification settings - Fork 40
Setup for using Banded Structures in BVP #124
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
Conversation
|
Conflicts |
7b2e9ba to
ee607ad
Compare
|
@ChrisRackauckas should we go ahead with this? If using CUDA / AMDGPU then we throw a fairly obvious error. Supporting it properly needs some custom cuda kernels (the state of CUSPARSE seems weird with most operations sending the arrays back to CPU) |
Yes. GPU needs a lot more. It needs custom kernel generation for how to construct the Jacobian, and then special CUSPARSE routines where we use the ilu0 (which is missing ldiv!'s). It needs more work, let's do that part separately. |
|
The CUDA part will unfortunately be a massive blocker if we want to Neural BVPs 😞 |
|
At least MultipleShooting is ready for CUDA? |
|
Need to test, but most likely no. It also specializes heavily on sparsity patterns |
|
For the purposes of Neural BVP if we avoid passing in the jacobian function and just use GMRES we should be good to go (I think) |
|
Yes that should be the case. Jacobian-free should be fine. Anything with a Jacobian involved will take more work (but be better overall). But that's why I had it as a separate project 😅 |
Uh oh!
There was an error while loading. Please reload this page.