Skip to content

Commit

Permalink
[onert] Fix weight deallocation to work on android (#3810)
Browse files Browse the repository at this point in the history
- Weight deallocation is enabled when `__ARM_NEON__` flag is exist
  - This flag is legacy and not supported on android
  - It is replaced by `__ARM_NEON`
- Use both `__ARM_NEON` and `__ARM_NEON__` flags to enable weight deallocation on android

ONE-DCO-1.0-Signed-off-by: JiHwan Yeo <[email protected]>
  • Loading branch information
periannath authored Aug 13, 2020
1 parent 7ad7c7f commit 155c095
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion runtime/onert/backend/cpu/ops/FullyConnectedLayer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,7 @@ void FullyConnectedLayer::prepare()
}
}

#if defined(__ARM_NEON__) && defined(USE_RUY_GEMV)
#if (defined(__ARM_NEON__) || defined(__ARM_NEON)) && defined(USE_RUY_GEMV)
// TODO This is workaround
// The only fc hybrid will use ruy kernel
if (_input->data_type() != OperandType::FLOAT32 ||
Expand Down

0 comments on commit 155c095

Please sign in to comment.