File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
pylops_mpi/basicoperators Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -496,6 +496,7 @@ def __init__(
496496 self ._col_comm = base_comm .Split (color = self ._col_id , key = self ._row_id )
497497
498498 self .A = A .astype (np .dtype (dtype ))
499+ if saveAt : self .At = A .T .conj ()
499500
500501 self .N = self ._col_comm .allreduce (A .shape [0 ])
501502 self .K = self ._row_comm .allreduce (A .shape [1 ])
@@ -571,8 +572,8 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
571572 for k in range (self ._P_prime ):
572573 Atemp = self .A .copy () if self ._col_id == k else np .empty_like (self .A )
573574 Xtemp = x_block .copy () if self ._row_id == k else np .empty_like (x_block )
574- self ._row_comm .Bcast (Atemp , root = k )
575- self ._col_comm .Bcast (Xtemp , root = k )
575+ self ._row_comm .bcast (Atemp , root = k )
576+ self ._col_comm .bcast (Xtemp , root = k )
576577 Y_local += ncp .dot (Atemp , Xtemp )
577578
578579 Y_local_unpadded = Y_local [:local_n , :local_m ]
You can’t perform that action at this time.
0 commit comments