Fix wrong relu implementation on non-ARM devices

JDAI-CV · Oct 10, 2019 · 55d7534 · 55d7534
1 parent 8a98287
commit 55d7534
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dabnn/layers/Relu.cpp b/dabnn/layers/Relu.cpp
@@ -20,7 +20,10 @@ void Relu::forward_impl() const {
     }
 #else
     float *ptr = static_cast<float *>(*data_mat);
-    FORZ(i, data_mat->total()) { *ptr = std::max(*ptr, 0.f); }
+    FORZ(i, data_mat->total()) {
+        *ptr = std::max(*ptr, 0.f);
+        ptr++;
+    }
 #endif  // __ARM_NEON
 }
 }  // namespace bnn