diff --git a/paddle/fluid/primitive/decomp_rule/decomp_vjp/details.h b/paddle/fluid/primitive/decomp_rule/decomp_vjp/details.h index c4ad2a474f9e6..833be0e82335d 100644 --- a/paddle/fluid/primitive/decomp_rule/decomp_vjp/details.h +++ b/paddle/fluid/primitive/decomp_rule/decomp_vjp/details.h @@ -1726,10 +1726,10 @@ void tile_grad(const Tensor& x, expand_shape_vec.size() <= 8) { auto repeat = repeat_times_data.back(); auto orig_size = - cast(out_grad_shape_vec.back() / repeat, DataType::INT32); + cast(out_grad_shape_vec.back() / repeat, DataType::INT64); size_t out_grad_last_index = out_grad_shape_vec.size() - 1; expand_shape_vec[out_grad_last_index] = - full({1}, repeat, DataType::INT32); + full({1}, repeat, DataType::INT64); expand_shape_vec.insert( expand_shape_vec.begin() + out_grad_shape_vec.size(), orig_size);