Added more timing. Fixed whitification.

This commit is contained in:
Guilhem Lavaux 2014-06-12 09:45:46 +02:00
parent 401ddc8a8b
commit a6b08dfe8f
7 changed files with 95 additions and 30 deletions

View file

@ -110,7 +110,7 @@ class LagrangianPerturbation(object):
div_phi2 -= (np.fft.irfftn( self._kdir(j)*self._kdir(i)*self.dhat / k2 ))**2
div_phi2 *= (self.N/self.L)**6
phi2_hat = np.fft.rfftn(div_phi2) * ((self.L/self.N)**3) / k2
phi2_hat = -np.fft.rfftn(div_phi2) * ((self.L/self.N)**3) / k2
self.cache['lpt2_potential'] = phi2_hat
del div_phi2
else: