diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/NEON/kernels/NEReorderKernel.cpp | 3 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEReorderKernel.h | 4 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEReorderLayer.cpp | 6 |
3 files changed, 12 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/NEReorderKernel.cpp b/src/core/NEON/kernels/NEReorderKernel.cpp index c09745604f..1a7f58bb08 100644 --- a/src/core/NEON/kernels/NEReorderKernel.cpp +++ b/src/core/NEON/kernels/NEReorderKernel.cpp @@ -21,6 +21,7 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#if defined(__aarch64__) #include "src/core/NEON/kernels/NEReorderKernel.h" #include "src/common/utils/Log.h" @@ -221,3 +222,5 @@ Status NEReorderKernel::validate(const ITensorInfo *input, const ITensorInfo *ou } } // namespace arm_compute + +#endif // defined(__aarch64__)
\ No newline at end of file diff --git a/src/core/NEON/kernels/NEReorderKernel.h b/src/core/NEON/kernels/NEReorderKernel.h index e744ff9e52..07908890f4 100644 --- a/src/core/NEON/kernels/NEReorderKernel.h +++ b/src/core/NEON/kernels/NEReorderKernel.h @@ -21,6 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#if defined(__aarch64__) + #ifndef ACL_SRC_CORE_NEON_KERNELS_NEREORDERKERNEL #define ACL_SRC_CORE_NEON_KERNELS_NEREORDERKERNEL @@ -91,3 +93,5 @@ private: } // namespace arm_compute #endif /* ACL_SRC_CORE_NEON_KERNELS_NEREORDERKERNEL */ + +#endif // defined(__aarch64__)
\ No newline at end of file diff --git a/src/runtime/NEON/functions/NEReorderLayer.cpp b/src/runtime/NEON/functions/NEReorderLayer.cpp index f9bbeba90c..53a7718902 100644 --- a/src/runtime/NEON/functions/NEReorderLayer.cpp +++ b/src/runtime/NEON/functions/NEReorderLayer.cpp @@ -21,6 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#if defined(__aarch64__) + #include "arm_compute/runtime/NEON/NEScheduler.h" #include "arm_compute/runtime/NEON/functions/NEReorderLayer.h" @@ -50,4 +52,6 @@ Status NEReorderLayer::validate(const ITensorInfo *input, const ITensorInfo *out return NEReorderKernel::validate(input, output, input_wf, output_wf); } -} // namespace arm_compute
\ No newline at end of file +} // namespace arm_compute + +#endif // defined(__aarch64__)
\ No newline at end of file |