Skip to content

Commit

Permalink
no need to memset dy_dtwo
Browse files Browse the repository at this point in the history
  • Loading branch information
njzjz authored Oct 11, 2023
1 parent 1d36d88 commit 5559d2a
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions source/lib/src/gpu/tabulate.cu
Original file line number Diff line number Diff line change
Expand Up @@ -798,9 +798,6 @@ void tabulate_fusion_se_a_grad_gpu(FPTYPE* dy_dem_x,
DPErrcheck(gpuDeviceSynchronize());
DPErrcheck(gpuMemset(dy_dem_x, 0, sizeof(FPTYPE) * nloc * nnei));
DPErrcheck(gpuMemset(dy_dem, 0, sizeof(FPTYPE) * nloc * nnei * 4));
if (two_embed != nullptr) {
DPErrcheck(gpuMemset(dy_dtwo, 0, sizeof(FPTYPE) * nloc * nnei));
}

tabulate_fusion_se_a_grad_fifth_order_polynomial<FPTYPE, MM, KK>
<<<nloc, KK * WARP_SIZE, sizeof(FPTYPE) * MM * last_layer_size>>>(
Expand Down

0 comments on commit 5559d2a

Please sign in to comment.