From 5e04b64fa0244239e7c476ebd846d20e9958d8d6 Mon Sep 17 00:00:00 2001
From: chengduoZH <zhaochengduo@163.com>
Date: Tue, 19 Dec 2017 16:19:29 +0800
Subject: [PATCH] refine elementwise

---
 paddle/operators/elementwise_op_function.h | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/paddle/operators/elementwise_op_function.h b/paddle/operators/elementwise_op_function.h
index 7ebfc7df8c..65484f318e 100644
--- a/paddle/operators/elementwise_op_function.h
+++ b/paddle/operators/elementwise_op_function.h
@@ -103,10 +103,12 @@ class MidWiseTransformIterator<T, platform::CPUDeviceContext> {
 
   MidWiseTransformIterator<T, platform::CPUDeviceContext>& operator++() {
     ++j_;
-    i_ = j_ / post_;
-    if (UNLIKELY(i_ == n_)) {
-      j_ = 0;
-      i_ = 0;
+    if (UNLIKELY(j_ == post_)) {
+      ++i_;
+      if (UNLIKELY(i_ == n_)) {
+        j_ = 0;
+        i_ = 0;
+      }
     }
     return *this;
   }
-- 
GitLab