[DCU] fix compiling error on DTK22.04, test=develop (#43999)

eb8fc759 · Qi Li · GitHub · b9640034 · eb8fc759
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

paddle/fluid/operators/class_center_sample_op.cu paddle/fluid/operators/class_center_sample_op.cu +6 -0

未找到文件。
--- 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 <typename Distance>
+  __host__ __device__ __forceinline__ self_type operator-(Distance n) const {
+    self_type ret(arr_, offset_ - n);
+    return ret;
+  }
+
  template <typename Distance>
  __host__ __device__ __forceinline__ reference operator[](Distance n) const {
    return *(*this + n);