< fix >

fix the memory release bug in unit test (test2-2.cpp and test2-3.cpp).

Author: sunliang98

source/module_pw/unittest/test2-2.cpp

@@ -55,10 +55,10 @@ int main(int argc,char **argv)
     GT = latvec.Inverse();
 	G  = GT.Transpose();
 	GGT = G * GT;
-    
+    complex<double> *tmp = NULL;
     if(myrank == 0)
     {
-        complex<double> *tmp = new complex<double> [nx*ny*nz];
+        tmp = new complex<double> [nx*ny*nz];
         for(int ix = 0 ; ix < nx ; ++ix)
         {
             for(int iy = 0 ; iy < ny ; ++iy)
@@ -80,7 +80,8 @@ int main(int argc,char **argv)
             }   
         }
         fftw_plan pp = fftw_plan_dft_3d(nx,ny,nz,(fftw_complex *) tmp, (fftw_complex *) tmp, FFTW_BACKWARD, FFTW_ESTIMATE);
-        fftw_execute(pp);     
+        fftw_execute(pp);    
+        fftw_destroy_plan(pp); 
 
         //output
         cout << "reference\n";
@@ -153,6 +154,9 @@ int main(int argc,char **argv)
     }
 
 
-    MPI_Barrier(MPI_COMM_WORLD);      
+    MPI_Barrier(MPI_COMM_WORLD);     
+    delete [] rhog;
+    delete [] rhor;
+    if(tmp!=NULL) delete []tmp; 
     return 0;
 }

source/module_pw/unittest/test2-3.cpp

@@ -55,10 +55,10 @@ int main(int argc,char **argv)
     GT = latvec.Inverse();
 	G  = GT.Transpose();
 	GGT = G * GT;
-    
+    complex<double> *tmp = NULL;
     if(myrank == 0)
     {
-        complex<double> *tmp = new complex<double> [nx*ny*nz];
+        tmp = new complex<double> [nx*ny*nz];
         for(int ix = 0 ; ix < nx ; ++ix)
         {
             for(int iy = 0 ; iy < ny ; ++iy)
@@ -81,7 +81,8 @@ int main(int argc,char **argv)
             }   
         }
         fftw_plan pp = fftw_plan_dft_3d(nx,ny,nz,(fftw_complex *) tmp, (fftw_complex *) tmp, FFTW_BACKWARD, FFTW_ESTIMATE);
-        fftw_execute(pp);     
+        fftw_execute(pp);  
+        fftw_destroy_plan(pp);    
 
         //output
         cout << "reference\n";
@@ -152,6 +153,8 @@ int main(int argc,char **argv)
     }
 
     MPI_Barrier(MPI_COMM_WORLD);
-
+    delete [] rhog;
+    delete [] rhor;
+    if(tmp!=NULL) delete []tmp;
     return 0;
 }