1 /*
2 * Copyright (c) 2018-2021 Arm Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24 #include "arm_compute/graph/Workload.h"
25
26 #include "arm_compute/graph/INode.h"
27 #include "arm_compute/graph/ITensorHandle.h"
28 #include "arm_compute/graph/nodes/PrintLayerNode.h"
29
30 #include "support/Cast.h"
31
32 namespace arm_compute
33 {
34 namespace graph
35 {
operator ()()36 void ExecutionTask::operator()()
37 {
38 TaskExecutor::get().execute_function(*this);
39 }
40
execute_task(ExecutionTask & task)41 void execute_task(ExecutionTask &task)
42 {
43 if(task.task)
44 {
45 task.task->run();
46 }
47 #ifdef ARM_COMPUTE_ASSERTS_ENABLED
48 else if(task.node->type() == NodeType::PrintLayer)
49 {
50 auto print_node = utils::cast::polymorphic_downcast<PrintLayerNode *>(task.node);
51 auto input_handle = print_node->input(0)->handle();
52 auto transform = print_node->transform();
53
54 input_handle->map(true);
55 ITensor *input = transform ? transform(&input_handle->tensor()) : &input_handle->tensor();
56 input->print(print_node->stream(), print_node->format_info());
57 input_handle->unmap();
58 }
59 #endif // ARM_COMPUTE_ASSERTS_ENABLED
60 }
61
prepare()62 void ExecutionTask::prepare()
63 {
64 if(task)
65 {
66 task->prepare();
67 }
68 }
69
TaskExecutor()70 TaskExecutor::TaskExecutor()
71 : execute_function(execute_task)
72 {
73 }
74
get()75 TaskExecutor &TaskExecutor::get()
76 {
77 static TaskExecutor executor;
78 return executor;
79 }
80 } // namespace graph
81 } // namespace arm_compute