diff --git a/paddle/fluid/operators/multinomial_op.h b/paddle/fluid/operators/multinomial_op.h index 05c96ffb4b0ff44bab25818274f488d17c3c2a21..cb63fc01972539460a21e8be990dec0996c31944 100644 --- a/paddle/fluid/operators/multinomial_op.h +++ b/paddle/fluid/operators/multinomial_op.h @@ -31,7 +31,7 @@ void MultinomialFunctor(T* out_data, const T* in_data, const int64_t num_samples, const bool replacement, const int64_t num_categories, const int64_t num_distributions) { - C = num_categories; + int64_t C = num_categories; T cumulative_probs[C]; std::uniform_real_distribution dist(0, 1);