Skip to content

Commit 3a81d01

Browse files
committed
broadcast A matrix in qr solve
1 parent 94d82a1 commit 3a81d01

File tree

1 file changed

+1
-0
lines changed
  • src/TiledArray/math/linalg/non-distributed

1 file changed

+1
-0
lines changed

src/TiledArray/math/linalg/non-distributed/qr.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ auto qr_solve(const ArrayA& A, const ArrayB& B,
4343
auto A_eig = detail::make_matrix(A);
4444
auto B_eig = detail::make_matrix(B);
4545
TA_LAPACK_ON_RANK_ZERO(qr_solve, world, A_eig, B_eig, cond);
46+
world.gop.broadcast_serializable(A_eig, 0);
4647
world.gop.broadcast_serializable(B_eig, 0);
4748
if (x_trange.rank() == 0) x_trange = B.trange();
4849
auto X = eigen_to_array<ArrayB>(world, x_trange, B_eig);

0 commit comments

Comments
 (0)