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

parallel computation #47

Open
jjdredd opened this issue Apr 29, 2022 · 2 comments
Open

parallel computation #47

jjdredd opened this issue Apr 29, 2022 · 2 comments

Comments

@jjdredd
Copy link

jjdredd commented Apr 29, 2022

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?

@flaport
Copy link
Owner

flaport commented Apr 29, 2022

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

@jjdredd
Copy link
Author

jjdredd commented Apr 29, 2022

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

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

2 participants