refine elementwise

5e04b64f · chengduoZH · 5ac72d95 · 5e04b64f
显示空白变更内容
内联并排

Showing with 6 addition and 4 deletion

paddle/operators/elementwise_op_function.h paddle/operators/elementwise_op_function.h +6 -4

未找到文件。
--- a/paddle/operators/elementwise_op_function.h
+++ b/paddle/operators/elementwise_op_function.h
@@ -103,11 +103,13 @@ class MidWiseTransformIterator<T, platform::CPUDeviceContext> {

  MidWiseTransformIterator<T, platform::CPUDeviceContext>& operator++() {
    ++j_;
-    i_ = j_ / post_;
+    if (UNLIKELY(j_ == post_)) {
+      ++i_;
      if (UNLIKELY(i_ == n_)) {
        j_ = 0;
        i_ = 0;
      }
+    }
    return *this;
  }