fix memory leaking in test_env_mat_a_mix.cc (#2596)

`new[]` should be used with `delete[]`. --------- Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Jun 7, 2023 · 4b822b8 · 4b822b8
1 parent a3c8980
commit 4b822b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/source/lib/tests/test_env_mat_a_mix.cc b/source/lib/tests/test_env_mat_a_mix.cc
@@ -553,7 +553,7 @@ TEST_F(TestEnvMatAMix, prod_cpu) {
       EXPECT_EQ(nmask[ii * nnei + jj], expected_nmask[ii * nnei + jj]);
     }
   }
-  free(nmask);
+  delete[] nmask;
 }
 
 TEST_F(TestEnvMatAMix, prod_cpu_equal_cpu) {
@@ -721,7 +721,7 @@ TEST_F(TestEnvMatAMix, prod_gpu_cuda) {
       EXPECT_EQ(nmask[ii * nnei + jj], expected_nmask[ii * nnei + jj]);
     }
   }
-  free(nmask);
+  delete[] nmask;
 }
 
 TEST_F(TestEnvMatAMix, prod_gpu_cuda_equal_cpu) {
@@ -939,7 +939,7 @@ TEST_F(TestEnvMatAMix, prod_gpu_rocm) {
       EXPECT_EQ(nmask[ii * nnei + jj], expected_nmask[ii * nnei + jj]);
     }
   }
-  free(nmask);
+  delete[] nmask;
 }
 
 TEST_F(TestEnvMatAMix, prod_gpu_rocm_equal_cpu) {