sigpy
sigpy copied to clipboard
Broadcasting arrays with MPI requires types to be known.
As of 85739eee0a2db4c14074b670e63e36de0cc5a8f2, MPI.COMPLEX is passed into Bcast. However, this is assuming the input array is complex. Ideally, there should be a dictionary that describes the relationship between data-types in numpy/cupy and the MPI types. I did try something akin to this but I got a dictionary error when trying to index complex types.