fix grammar

deepmodeling · Oct 7, 2023 · 7fbea3d · 7fbea3d
1 parent 6a0fd03
commit 7fbea3d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/lib/src/gpu/tabulate.cu b/source/lib/src/gpu/tabulate.cu
@@ -429,7 +429,7 @@ __global__ void tabulate_fusion_se_a_grad_grad_fifth_order_polynomial(
      * dz_xx`.
      *
      * If `enable_se_atten` is true, `res` will be `res * t + res`, and `res'`
-     * becomes `(res_grad * t + res_grad) * dz_xx`.
+     * will become `(res_grad * t + res_grad) * dz_xx`.
      */
     for (int kk = 0; kk < MTILE; kk++) {
       int em_index = block_idx * nnei * MTILE + ii * MTILE + kk;

diff --git a/source/lib/src/tabulate.cc b/source/lib/src/tabulate.cc
@@ -325,7 +325,7 @@ void deepmd::tabulate_fusion_se_a_grad_grad_cpu(FPTYPE* dz_dy,
          * as: `var_grad * dz_xx`.
          *
          * If `enable_se_atten` is true, `var` will be `var * t + var`, and
-         * `var'` becomes `(var_grad * t + var_grad) * dz_xx`.
+         * `var'` will be `(var_grad * t + var_grad) * dz_xx`.
          */
         if (unloop) {
           dz_dy[ii * last_layer_size * 4 + 0 * last_layer_size + kk] +=