Fix possible memory leak.

RMGDFT · Jan 27, 2023 · b314051 · b314051
1 parent d393d3d
commit b314051
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/RMG/Common/GetAugRho.cpp b/RMG/Common/GetAugRho.cpp
@@ -57,10 +57,11 @@ template <typename KpointType> void GetAugRho(Kpoint<KpointType> **Kpts, double
 
     int factor = ct.noncoll_factor * ct.noncoll_factor;
     for(int idx = 0;idx < pbasis*factor;idx++) augrho[idx] = 0.0;
+    if(ct.norm_conserving_pp) return;
+
     double *taugrho = new double[pbasis*factor]();
 
 
-    if(ct.norm_conserving_pp) return;
 
     std::complex<double> *product = new std::complex<double>[max_product * factor];
     std::complex<double> *product_tem = new std::complex<double>[max_product * factor];