SciML / SciML/DiffEqGPU.jl

matrix differential equation support

Open
#173 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Julia
Stars
327
Forks
42
Avg merge
14h 13m
Merged PRs (30d)
34

Description

The documentation says CuArray u0 are supported.
But only one dimensional arrays are supported because of lines like:
https://github.com/SciML/DiffEqGPU.jl/blob/8489a3e74f9179aafddb38912bedd9321fadd68b/src/DiffEqGPU.jl#L345

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:
https://github.com/SciML/DiffEqGPU.jl/blob/8489a3e74f9179aafddb38912bedd9321fadd68b/src/DiffEqGPU.jl#L24

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start in src/DiffEqGPU.jl at the referenced lines around 345 and 24. Trace how CuArray initial conditions are concatenated and how the kernels index them, then generalize both paths from one-dimensional inputs to N-dimensional inputs. Done means matrix and higher-dimensional differential-equation initial conditions are accepted and indexed across the final dimension.

Written by the indexing model from the issue text.

Assessment

Tech stack
julia
Domain
hpc
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.