You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Issue: Chunking along the size of MPI isn't always optimal. It can lead to deadlocks of sending chunks to proccesses due to each node expecting a chunk while the chunk operation likes splitting some things along smaller grouping. (E.G. a torch.Tensor(8,x,y,z):chunk(6,1) will return 4 chunks instead of 6. THis can be problematic if you are chunking on the number of MPI nodes.