fix

Author: OutisLi

source/lib/tests/test_tabulate_se_t_tebd.cc

@@ -689,13 +689,12 @@ TEST_F(TestTabulateSeTTebd, tabulate_fusion_se_t_tebd_cpu) {
 
 TEST_F(TestTabulateSeTTebd, tabulate_fusion_se_t_tebd_grad_cpu) {
   std::vector<double> dy_dem_x(em_x.size());
-  std::vector<double> dy_dem(em.size());
   std::vector<double> dy(expected_xyz_scatter.begin(),
                          expected_xyz_scatter.end());
 
   deepmd::tabulate_fusion_se_t_tebd_grad_cpu<double>(
-      &dy_dem_x[0], &dy_dem[0], &table[0], &table_info[0], &em_x[0], &em[0],
-      &dy[0], nloc, nnei_i, nnei_j, last_layer_size);
+      &dy_dem_x[0], &table[0], &table_info[0], &em_x[0], &em[0], &dy[0], nloc,
+      nnei_i, nnei_j, last_layer_size);
 
   EXPECT_EQ(dy_dem_x.size(), nloc * nnei_i * nnei_j);
   EXPECT_EQ(dy_dem_x.size(), expected_dy_dem_x.size());
@@ -739,31 +738,27 @@ TEST_F(TestTabulateSeTTebd, tabulate_fusion_se_t_tebd_gpu) {
 
 TEST_F(TestTabulateSeTTebd, tabulate_fusion_se_t_tebd_grad_gpu) {
   std::vector<double> dy_dem_x(em_x.size(), 0.0);
-  std::vector<double> dy_dem(em.size(), 0.0);
   std::vector<double> dy(expected_xyz_scatter.begin(),
                          expected_xyz_scatter.end());
 
-  double *dy_dem_x_dev = NULL, *dy_dem_dev = NULL, *table_dev = NULL,
-         *table_info_dev = NULL, *em_x_dev = NULL, *em_dev = NULL,
-         *dy_dev = NULL;
+  double *dy_dem_x_dev = NULL, *table_dev = NULL, *table_info_dev = NULL,
+         *em_x_dev = NULL, *em_dev = NULL, *dy_dev = NULL;
 
   deepmd::malloc_device_memory_sync(dy_dem_x_dev, dy_dem_x);
-  deepmd::malloc_device_memory_sync(dy_dem_dev, dy_dem);
   deepmd::malloc_device_memory_sync(table_dev, table);
   deepmd::malloc_device_memory_sync(table_info_dev, table_info);
   deepmd::malloc_device_memory_sync(em_x_dev, em_x);
   deepmd::malloc_device_memory_sync(em_dev, em);
   deepmd::malloc_device_memory_sync(dy_dev, dy);
 
   deepmd::tabulate_fusion_se_t_tebd_grad_gpu<double>(
-      dy_dem_x_dev, dy_dem_dev, table_dev, table_info_dev, em_x_dev, em_dev,
-      dy_dev, nloc, nnei_i, nnei_j, last_layer_size);
+      dy_dem_x_dev, table_dev, table_info_dev, em_x_dev, em_dev, dy_dev, nloc,
+      nnei_i, nnei_j, last_layer_size);
 
   deepmd::memcpy_device_to_host(&dy_dem_x[0], dy_dem_x_dev,
                                 dy_dem_x.size() * sizeof(double));
 
   deepmd::delete_device_memory(dy_dem_x_dev);
-  deepmd::delete_device_memory(dy_dem_dev);
   deepmd::delete_device_memory(table_dev);
   deepmd::delete_device_memory(table_info_dev);
   deepmd::delete_device_memory(em_x_dev);