diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-01-20 18:16:47 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-01-21 16:23:55 +0000 |
commit | 0094c023038cbb353f60e96d1301b4bb25c9e382 (patch) | |
tree | 18d0c8e312c16f82e272b959b9aca92c655a5b15 /src/runtime/NEON/functions | |
parent | f47f718e87889b7d501a48e1d86b5d12b2e3f54c (diff) | |
download | ComputeLibrary-0094c023038cbb353f60e96d1301b4bb25c9e382.tar.gz |
Add window parameter to scheduler
To support window configured outside of
kernels, the parameter is added to scheduler.
The existing operators pass window from
kernels to preserve the current behavior.
Partial Implements: COMPMID-4003
Change-Id: I4514e12fb5b0c60e4adfc1a51e53a8a5de356a0d
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4892
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions')
-rw-r--r-- | src/runtime/NEON/functions/NELogical.cpp | 8 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp | 4 |
2 files changed, 6 insertions, 6 deletions
diff --git a/src/runtime/NEON/functions/NELogical.cpp b/src/runtime/NEON/functions/NELogical.cpp index 2c9ebd5f29..190998b042 100644 --- a/src/runtime/NEON/functions/NELogical.cpp +++ b/src/runtime/NEON/functions/NELogical.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -65,7 +65,7 @@ Status NELogicalAnd::validate(const ITensorInfo *input1, const ITensorInfo *inpu void NELogicalAnd::run() { - NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->pack); + NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->kernel->window(), _impl->pack); } struct NELogicalOr::Impl : public LogicalArgs @@ -98,7 +98,7 @@ Status NELogicalOr::validate(const ITensorInfo *input1, const ITensorInfo *input void NELogicalOr::run() { - NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->pack); + NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->kernel->window(), _impl->pack); } struct NELogicalNot::Impl : public LogicalArgs @@ -130,6 +130,6 @@ Status NELogicalNot::validate(const ITensorInfo *input, const ITensorInfo *outpu void NELogicalNot::run() { - NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->pack); + NEScheduler::get().schedule_op(_impl->kernel.get(), Window::DimY, _impl->kernel->window(), _impl->pack); } } // namespace arm_compute diff --git a/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp index 2600e2b946..427cd2eb77 100644 --- a/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEPoolingAssemblyDispatch.cpp @@ -78,11 +78,11 @@ void NEPoolingAssemblyDispatch::run(ITensorPack &tensors) if(_is_global_pooling_layer) { - NEScheduler::get().schedule_op(_kernel.get(), Window::DimX, tensors); + NEScheduler::get().schedule_op(_kernel.get(), Window::DimX, _kernel->window(), tensors); } else { - NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, tensors); + NEScheduler::get().schedule_op(_kernel.get(), Window::DimY, _kernel->window(), tensors); } } } // namespace experimental |