Skip to content

Commit e2f9cae

Browse files
authored
[BIG tensor] Fix cuda error in addmm (#74273)
1 parent 97f18d5 commit e2f9cae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/kernels/impl/addmm_grad_kernel_impl.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ void AddmmGradKernel(const Context& dev_ctx,
9797
in_dims = {1, input.dims()[0]};
9898
input_grad->Resize(in_dims);
9999
}
100-
int total_elems = 0;
100+
int64_t total_elems = 0;
101101

102102
VLOG(3) << "alpha: " << alpha << " beta: " << beta;
103103

0 commit comments

Comments
 (0)