There was an error while loading. Please reload this page.
1 parent 0c8c834 commit 33ffe85Copy full SHA for 33ffe85
paddle/phi/kernels/cpu/trunc_grad_kernel.cc
@@ -25,7 +25,7 @@ void TruncGradKernel(const Context& dev_ctx,
25
DenseTensor* in_grad) {
26
T* dx_data = dev_ctx.template Alloc<T>(in_grad);
27
28
- int64_t numel = static_cast<int>(in_grad->numel());
+ int64_t numel = in_grad->numel();
29
memset(dx_data, 0.0, numel * sizeof(T));
30
}
31
0 commit comments