diff --git a/paddle/fluid/operators/class_center_sample_op.cu b/paddle/fluid/operators/class_center_sample_op.cu index 236f783017dcccde80dc94840a7509190912089b..a0642694843e8afe73a9947876ee93648d402449 100644 --- a/paddle/fluid/operators/class_center_sample_op.cu +++ b/paddle/fluid/operators/class_center_sample_op.cu @@ -220,6 +220,12 @@ class NotEqualToPreviousAdjacentIterator { return ret; } + template + __host__ __device__ __forceinline__ self_type operator-(Distance n) const { + self_type ret(arr_, offset_ - n); + return ret; + } + template __host__ __device__ __forceinline__ reference operator[](Distance n) const { return *(*this + n);