!5049 [MS][LITE][Develop]fix_arm32_comple

Merge pull request !5049 from sunsuodong/fix_arm32_comple
5 years ago · 6c309d8f62
parent 33a562de3d f50a235cee
commit 6c309d8f62
1 changed files with 1 additions and 1 deletions
--- a/mindspore/lite/nnacl/fp32/prelu.c
+++ b/mindspore/lite/nnacl/fp32/prelu.c
@ -25,7 +25,7 @@ void PRelu(float *input, float *output, PReluParameter *prelu_param_, int task_i
  for (int j = task_id; j < prelu_param_->tile_block_; j += prelu_param_->op_parameter_.thread_num_) {
    float *input_ptr = input + j * TILE_NUM * channel_num;
    float *output_ptr = input_ptr;
-#ifdef ENABLE_NEON
+#ifdef ENABLE_ARM64
    for (int i = 0; i < c4; i++) {
      int c_offset = i * C4NUM;
      float32x4_t slope_value = vld1q_f32(negetive_slope_value + c_offset);