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

matrix differential equation support #173

Open
ArnoStrouwen opened this issue Aug 11, 2022 · 0 comments
Open

matrix differential equation support #173

ArnoStrouwen opened this issue Aug 11, 2022 · 0 comments

Comments

@ArnoStrouwen
Copy link
Member

The documentation says CuArray u0 are supported.
But only one dimensional arrays are supported because of lines like:

u0 = reduce(hcat, Array(probs[i].u0) for i in 1:length(I))

To generalize this to N dimensions the u0 should concatenate across an additional dimension.
And the kernels should then index over this last dimension, instead of assuming there are only 2:

@views @inbounds f(du[:, i], u[:, i], p[:, i], t)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant