Skip to content

Commit d001383

Browse files
stevengjfredrikekre
authored andcommitted
pass rng recursively in lobpcg!
1 parent d89d1c8 commit d001383

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/lobpcg.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -943,7 +943,7 @@ function lobpcg(A, B, largest::Bool, X0, nev::Int;
943943
iterator = LOBPCGIterator(A, B, largest, X, nev, P, C)
944944

945945
r = EmptyLOBPCGResults(X, nev, tol, maxiter)
946-
rnext = lobpcg!(iterator, log=log, tol=tol, maxiter=maxiter, not_zeros=not_zeros)
946+
rnext = lobpcg!(iterator, rng=rng, log=log, tol=tol, maxiter=maxiter, not_zeros=not_zeros)
947947
append!(r, rnext, 0)
948948
converged_x = sizeX
949949
while converged_x < nev
@@ -952,13 +952,13 @@ function lobpcg(A, B, largest::Bool, X0, nev::Int;
952952
update!(iterator.constr!, iterator.XBlocks.block[:, 1:cutoff], iterator.XBlocks.B_block[:, 1:cutoff])
953953
X[:, 1:sizeX-cutoff] .= X[:, cutoff+1:sizeX]
954954
rand!(rng, X[:, cutoff+1:sizeX])
955-
rnext = lobpcg!(iterator, log=log, tol=tol, maxiter=maxiter, not_zeros=true)
955+
rnext = lobpcg!(iterator, rng=rng, log=log, tol=tol, maxiter=maxiter, not_zeros=true)
956956
append!(r, rnext, converged_x, sizeX-cutoff)
957957
converged_x += sizeX-cutoff
958958
else
959959
update!(iterator.constr!, iterator.XBlocks.block, iterator.XBlocks.B_block)
960960
rand!(rng, X)
961-
rnext = lobpcg!(iterator, log=log, tol=tol, maxiter=maxiter, not_zeros=true)
961+
rnext = lobpcg!(iterator, rng=rng, log=log, tol=tol, maxiter=maxiter, not_zeros=true)
962962
append!(r, rnext, converged_x)
963963
converged_x += sizeX
964964
end

0 commit comments

Comments
 (0)