Skip to content

Commit 2d525d0

Browse files
authored
Updated ArmCompute to 2021.2 (opencv#78)
1 parent f039574 commit 2d525d0

File tree

4 files changed

+15
-8
lines changed

4 files changed

+15
-8
lines changed

modules/arm_plugin/src/arm_ie_scheduler.cpp

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,11 @@ std::uint32_t IEScheduler::num_threads() const {
1717
return parallel_get_max_threads();
1818
}
1919

20-
void IEScheduler::Schedule(arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints, arm_compute::ITensorPack& tensors) {
20+
void IEScheduler::Schedule(arm_compute::ICPPKernel* kernel,
21+
const arm_compute::IScheduler::Hints& hints,
22+
const arm_compute::Window& max_window,
23+
arm_compute::ITensorPack& tensors) {
2124
IE_ASSERT(kernel != nullptr);
22-
const auto& max_window = kernel->window();
2325

2426
auto splitDimension = hints.split_dimension();
2527

@@ -73,13 +75,14 @@ void IEScheduler::Schedule(arm_compute::ICPPKernel* kernel, const arm_compute::I
7375

7476
void IEScheduler::schedule(arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints) {
7577
arm_compute::ITensorPack tensors;
76-
Schedule(kernel, hints, tensors);
78+
Schedule(kernel, hints, kernel->window(), tensors);
7779
}
7880

7981
void IEScheduler::schedule_op(arm_compute::ICPPKernel* kernel,
8082
const arm_compute::IScheduler::Hints& hints,
83+
const arm_compute::Window& window,
8184
arm_compute::ITensorPack& tensors) {
82-
Schedule(kernel, hints, tensors);
85+
Schedule(kernel, hints, window, tensors);
8386
}
8487

8588
void IEScheduler::run_workloads(std::vector<arm_compute::IScheduler::Workload>& workloads) {

modules/arm_plugin/src/arm_ie_scheduler.hpp

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,15 @@ struct IEScheduler final : public arm_compute::IScheduler {
1414
~IEScheduler() override;
1515
void set_num_threads(unsigned int num_threads) override;
1616
std::uint32_t num_threads() const override;
17-
void Schedule(arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints, arm_compute::ITensorPack& tensors);
17+
void Schedule(arm_compute::ICPPKernel* kernel,
18+
const arm_compute::IScheduler::Hints& hints,
19+
const arm_compute::Window& max_window,
20+
arm_compute::ITensorPack& tensors);
1821
void schedule(arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints) override;
1922
void schedule_op(arm_compute::ICPPKernel* kernel,
2023
const arm_compute::IScheduler::Hints& hints,
21-
arm_compute::ITensorPack& tensors);
24+
const arm_compute::Window& window,
25+
arm_compute::ITensorPack& tensors) override;
2226
void run_workloads(std::vector<arm_compute::IScheduler::Workload>& workloads) override;
2327
};
2428
} // namespace ArmPlugin

modules/arm_plugin/third-party-programs.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ terms are listed below.
1515
-------------------------------------------------------------
1616

1717
1. ComputeLibrary
18-
Copyright (c) 2017-2020 Arm Limited
18+
Copyright (c) 2017-2021 Arm Limited
1919

2020
MIT License
2121

Submodule ComputeLibrary updated 1472 files

0 commit comments

Comments
 (0)