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

Enable prange over nodes for Numba backend #543

Open
maedoc opened this issue Apr 1, 2022 · 1 comment
Open

Enable prange over nodes for Numba backend #543

maedoc opened this issue Apr 1, 2022 · 1 comment
Labels
area/scilib enhancement New feature or request good first issue Good for newcomers

Comments

@maedoc
Copy link
Member

maedoc commented Apr 1, 2022

Describe the new feature or enhancement

Provide parallelization at the node level in simulations.

Describe your proposed implementation

The Numba backend has a for loop over nodes in the network here:

    for i in range(weights.shape[0]):

Enabling the backend to switch to numba.prange would enable parallelization over nodes, and the switch would be an option passed in by the user/client of the backend class when instantiating it.

Describe possible alternatives

  • Numba supports some automatic parallelization with the gufunc mechanism, but it would require a different template structure from the current backends
  • Numba cuda backend could do this as well, but not yet mature enough to be used

Additional comments

This was asked about in the TVB list this morning, https://groups.google.com/g/tvb-users/c/GovsAb-xc1k/m/vCTOQcd_FAAJ

@maedoc maedoc added enhancement New feature or request good first issue Good for newcomers area/scilib labels Apr 1, 2022
@Bikisome
Copy link

hello @maedoc sir i made a pull request on this issue so please review it sir

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
area/scilib enhancement New feature or request good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

2 participants