fdtd icon indicating copy to clipboard operation
fdtd copied to clipboard

parallel computation

Open jjdredd opened this issue 4 years ago • 2 comments

Hello, Thanks for your work. Is there a way to run this code in parallel on cpu? I assume that it's automatically parallelized if the cuda backend is used. Is this correct?

jjdredd avatar Apr 29 '22 06:04 jjdredd

Hi @jjdredd ,

Currently the simulator supports computations on a CPU (numpy backend) or computations on a GPU (torch backend). However, further parallelization (for example using MPI) is not supported.

That said, it should be fairly trivial to create a dask backend (see backend.py). I would happily accept a PR for that :)

Best, Floris

flaport avatar Apr 29 '22 12:04 flaport

Thanks. I'll take a look. didn't know about dask.

jjdredd avatar Apr 29 '22 19:04 jjdredd