jan/meanas
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

comment updates

Browse Source
This commit is contained in:
Jan Petykiewicz 2023-05-23 12:54:18 -07:00
parent 2c16c3c9ab
commit be620f7137
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
meanas/fdfd/bloch.py
View File

@ -596,8 +596,9 @@ def eigsolve(
E_signed = real(trace(ZtAZU)) E_signed = real(trace(ZtAZU))
sgn = numpy.sign(E_signed) sgn = numpy.sign(E_signed)
E = numpy.abs(E_signed) E = numpy.abs(E_signed)
G = (AZ @ U - Z @ U @ ZtAZU) * sgn # G = AZU projected onto the space orthonormal to Z
# via (1 - ZUZt) # G = AZU projected onto the space orthonormal to Z via (1 - ZUZt)
G = (AZ @ U - Z @ U @ ZtAZU) * sgn
if i > 0 and abs(E - prev_E) < tolerance * 0.5 * (E + prev_E + 1e-7): if i > 0 and abs(E - prev_E) < tolerance * 0.5 * (E + prev_E + 1e-7):
logger.info( logger.info(
@ -608,7 +609,7 @@ def eigsolve(
break break
KG = scipy_iop @ G # Preconditioned steepest descent direction KG = scipy_iop @ G # Preconditioned steepest descent direction
traceGtKG = _rtrace_AtB(G, KG) # traceGtKG = _rtrace_AtB(G, KG)
if prev_traceGtKG == 0 or i % reset_iters == 0: if prev_traceGtKG == 0 or i % reset_iters == 0:
logger.info('CG reset') logger.info('CG reset')