fix compile

deepmodeling · Nov 14, 2024 · 0edfdc0 · 0edfdc0
1 parent 0e2f25b
commit 0edfdc0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/source/module_hamilt_pw/hamilt_pwdft/kernels/test/force_op_test.cpp b/source/module_hamilt_pw/hamilt_pwdft/kernels/test/force_op_test.cpp
@@ -2899,6 +2899,7 @@ TEST_F(TestSrcPWForceMultiDevice, cal_force_nl_op_cpu)
                                                                atom_na.data(),
                                                                tpiba,
                                                                wg.data(),
+                                                               true,
                                                                ekb.data(),
                                                                qq_nt.data(),
                                                                deeq.data(),
@@ -2990,6 +2991,7 @@ TEST_F(TestSrcPWForceMultiDevice, cal_force_nl_op_gpu)
                                                                d_atom_na,
                                                                tpiba,
                                                                d_wg,
+                                                               true,
                                                                d_ekb,
                                                                d_qq_nt,
                                                                d_deeq,

diff --git a/source/module_hamilt_pw/hamilt_pwdft/kernels/test/stress_op_test.cpp b/source/module_hamilt_pw/hamilt_pwdft/kernels/test/stress_op_test.cpp
@@ -102,6 +102,7 @@ TEST(TestSrcPWStressMultiDevice, cal_stress_nl_op_cpu)
                                                                 atom_nh.data(),
                                                                 atom_na.data(),
                                                                 wg.data(),
+                                                                true,
                                                                 ekb.data(),
                                                                 qq_nt.data(),
                                                                 deeq.data(),
@@ -275,6 +276,7 @@ TEST(TestSrcPWStressMultiDevice, cal_stress_nl_op_gpu)
                                                                 d_atom_nh,
                                                                 d_atom_na,
                                                                 d_wg,
+                                                                true,
                                                                 d_ekb,
                                                                 d_qq_nt,
                                                                 d_deeq,