summary of the add veff_op_test

Author: A-006

source/module_hamilt_pw/hamilt_pwdft/kernels/test/veff_op_test.cpp

@@ -168,18 +168,20 @@ TEST_F(TestModuleHamiltVeff, veff_pw_spin_op_gpu_batch)
     std::vector<std::complex<double>> out1_spin(out_spin.size(), std::complex<double>(0, 0));
     std::vector<std::complex<double>> expected_out1_spin(out_spin.size(), std::complex<double>(0, 0));
 
-    std::vector<std::complex<double>> res = out_spin;
-    std::vector<std::complex<double>> res1 = out1_spin;
-    res.insert(res.end(), res.begin(), res.end());
-    res1.insert(res1.end(), res1.begin(), res1.end());
+    std::vector<std::complex<double>> res_batch = out_spin;
+    std::vector<std::complex<double>> res1_batch = out1_spin;
+    std::vector<std::complex<double>> expected_out_spin_batch = expected_out_spin;
+    std::vector<std::complex<double>> expected_out1_spin_batch = expected_out1_spin;
+    res_batch.insert(res_batch.end(), res_batch.begin(), res_batch.end());
+    res1_batch.insert(res1_batch.end(), res1_batch.begin(), res1_batch.end());
     double* d_in = NULL;
     std::complex<double>* d_res = NULL, * d_res1 = NULL;
     resize_memory_double_op()(d_in, in_spin.size());
-    resize_memory_complex_op()(d_res, res.size());
-    resize_memory_complex_op()(d_res1, res1.size());
+    resize_memory_complex_op()(d_res, res_batch.size());
+    resize_memory_complex_op()(d_res1, res1_batch.size());
     syncmem_double_h2d_op()(d_in, in_spin.data(), in_spin.size());
-    syncmem_complex_h2d_op()(d_res, res.data(), res.size());
-    syncmem_complex_h2d_op()(d_res1, res1.data(), res1.size());
+    syncmem_complex_h2d_op()(d_res, res_batch.data(), res_batch.size());
+    syncmem_complex_h2d_op()(d_res1, res1_batch.data(), res1_batch.size());
 
     const double * in_[4];
     for (int ii = 0; ii < 4; ii++) {
@@ -188,13 +190,13 @@ TEST_F(TestModuleHamiltVeff, veff_pw_spin_op_gpu_batch)
 
     veff_gpu_op()(gpu_ctx, this->size, d_res, d_res1, in_ , 2);
 
-    syncmem_complex_d2h_op()(res.data(), d_res, res.size());
-    syncmem_complex_d2h_op()(res1.data(), d_res1, res1.size());
-    expected_out_spin.insert(expected_out_spin.end(), expected_out_spin.begin(), expected_out_spin.end());
-    expected_out1_spin.insert(expected_out1_spin.end(), expected_out1_spin.begin(), expected_out1_spin.end());
-    for (int ii = 0; ii < res.size(); ii++) {
-        EXPECT_LT(fabs(res[ii] - expected_out_spin[ii]), 7.5e-5);
-        EXPECT_LT(fabs(res1[ii] - expected_out1_spin[ii]), 6e-5);
+    syncmem_complex_d2h_op()(res_batch.data(), d_res, res_batch.size());
+    syncmem_complex_d2h_op()(res1_batch.data(), d_res1, res1_batch.size());
+    expected_out_spin_batch.insert(expected_out_spin_batch.end(), expected_out_spin_batch.begin(), expected_out_spin_batch.end());
+    expected_out1_spin_batch.insert(expected_out1_spin_batch.end(), expected_out1_spin_batch.begin(), expected_out1_spin_batch.end());
+    for (int ii = 0; ii < res_batch.size(); ii++) {
+        EXPECT_LT(fabs(res_batch[ii] - expected_out_spin_batch[ii]), 7.5e-5);
+        EXPECT_LT(fabs(res1_batch[ii] - expected_out1_spin_batch[ii]), 6e-5);
     }
     delete_memory_double_op()(d_in);
     delete_memory_complex_op()(d_res);