Python Eigensolver.estimate_memory Examples

Example #1

File: davidson.py Project: ryancoleman/lotsofcoresbook2code

 def estimate_memory(self, mem, wfs):
     Eigensolver.estimate_memory(self, mem, wfs)
     nbands = wfs.bd.nbands
     mem.subnode("H_nn", nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode("S_nn", nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode("H_2n2n", 4 * nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode("S_2n2n", 4 * nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode("eps_2n", 2 * nbands * mem.floatsize)

Example #2

File: davidson.py Project: thonmaker/gpaw

 def estimate_memory(self, mem, wfs):
     Eigensolver.estimate_memory(self, mem, wfs)
     nbands = wfs.bd.nbands
     mem.subnode('H_nn', nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode('S_nn', nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode('H_2n2n', 4 * nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode('S_2n2n', 4 * nbands * nbands * mem.itemsize[wfs.dtype])
     mem.subnode('eps_2n', 2 * nbands * mem.floatsize)

Example #3

File: davidson.py Project: yihsuanliu/gpaw

 def estimate_memory(self, mem, gd, dtype, mynbands, nbands):
     Eigensolver.estimate_memory(self, mem, gd, dtype, mynbands, nbands)
     itemsize = mem.itemsize[dtype]
     mem.subnode('H_nn', nbands * nbands * mem.itemsize[dtype])
     mem.subnode('S_nn', nbands * nbands * mem.itemsize[dtype])
     mem.subnode('H_2n2n', 4 * nbands * nbands * mem.itemsize[dtype])
     mem.subnode('S_2n2n', 4 * nbands * nbands * mem.itemsize[dtype])
     mem.subnode('eps_2n', 2 * nbands * mem.floatsize)

Example #4

File: cg.py Project: yihsuanliu/gpaw

 def estimate_memory(self, mem, gd, dtype, mynbands, nbands):
     Eigensolver.estimate_memory(self, mem, gd, dtype, mynbands, nbands)
     gridmem = gd.bytecount(dtype)
     mem.subnode('phi_G', gridmem)
     mem.subnode('phi_old_G', gridmem)