diff --git a/dabnn/layers/Relu.cpp b/dabnn/layers/Relu.cpp index 4d6dbeb..4d82008 100644 --- a/dabnn/layers/Relu.cpp +++ b/dabnn/layers/Relu.cpp @@ -20,7 +20,10 @@ void Relu::forward_impl() const { } #else float *ptr = static_cast(*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