Refactoring of communication
: separate mpi4py
-wrappers from DNDarrays
#1243
Labels
communication
: separate mpi4py
-wrappers from DNDarrays
#1243
Currently, the code wrapping
mpi4py
fortorch.Tensor
's is contained inheat/core/communication.py
and not fully separated from constructions related to theDNDarray
-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:
mpi4py
fortorch.Tensor
's becomes an own module of heatheat/core/communication.py
this module can be usedDiclaimer: feel free to discuss other ideas for refactoring and/or that status-quo does not need changes
The text was updated successfully, but these errors were encountered: