heat icon indicating copy to clipboard operation
heat copied to clipboard

Refactoring of `communication`: separate `mpi4py`-wrappers from `DNDarrays`

Open mrfh92 opened this issue 1 year ago • 2 comments

Currently, the code wrapping mpi4py for torch.Tensor's is contained in heat/core/communication.py and not fully separated from constructions related to the DNDarray-class (e.g., computation of chunk sizes is also introduced there).

I would suggest to separate this clearly, which may also make #482 and similar work easier:

  • code wrapping mpi4py for torch.Tensor's becomes an own module of heat
  • in heat/core/communication.py this module can be used

Diclaimer: feel free to discuss other ideas for refactoring and/or that status-quo does not need changes

mrfh92 avatar Oct 13 '23 12:10 mrfh92

This issue is stale because it has been open for 60 days with no activity.

github-actions[bot] avatar Jan 22 '24 02:01 github-actions[bot]