Skip to content

Commit

Permalink
chore: remove incorrect memset TODOs (deepmodeling#3600)
Browse files Browse the repository at this point in the history
Fix deepmodeling#3556.

Signed-off-by: Jinzhe Zeng <[email protected]>
(cherry picked from commit 625e893)
  • Loading branch information
njzjz committed Apr 6, 2024
1 parent 164288f commit fc8a4fa
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions source/lib/src/gpu/prod_env_mat.cu
Original file line number Diff line number Diff line change
Expand Up @@ -486,7 +486,6 @@ __global__ void compute_env_mat_a(FPTYPE* em,
std[type[bid] * ndescrpt + idx_value + ii];
}
} else {
// TODO: move it to the memset.
row_descript[idx_value] -= avg[type[bid] * ndescrpt + idx_value] /
std[type[bid] * ndescrpt + idx_value];
}
Expand Down Expand Up @@ -562,7 +561,6 @@ __global__ void compute_env_mat_r(FPTYPE* em,
row_em[idx_value] = (dd - avg[type[bid] * ndescrpt + idx_value]) /
std[type[bid] * ndescrpt + idx_value];
} else {
// TODO: move it to the memset.
row_em[idx_value] -= avg[type[bid] * ndescrpt + idx_value] /
std[type[bid] * ndescrpt + idx_value];
}
Expand Down

0 comments on commit fc8a4fa

Please sign in to comment.