diff --git a/paddle/operators/scatter.h b/paddle/operators/scatter.h index 714c022c02d15fc4c2a22c44747273c17fc5a63d..6b542675c291607b35f180123cf42fee6a783a85 100644 --- a/paddle/operators/scatter.h +++ b/paddle/operators/scatter.h @@ -75,12 +75,12 @@ void ScatterUpdate(const platform::Place& place, auto dst_dims = output->dims(); // check src shape and dst shape should match - for (size_t i = 1; i < src_dims.size(); i++) + for (int i = 1; i < src_dims.size(); i++) PADDLE_ENFORCE(src_dims[i] == dst_dims[i]); // slice size size_t slice_size = 1; - for (size_t i = 0; i < src_dims.size(); ++i) slice_size *= src_dims[i]; + for (int i = 0; i < src_dims.size(); ++i) slice_size *= src_dims[i]; if (platform::is_cpu_place(place)) { CPUScatterUpdate(src, index->data(), index_size, output);