Compute Library
 21.11
Workload.cpp
Go to the documentation of this file.
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  */
25 
29 
30 #include "support/Cast.h"
31 
32 namespace arm_compute
33 {
34 namespace graph
35 {
37 {
39 }
40 
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 
63 {
64  if(task)
65  {
66  task->prepare();
67  }
68 }
69 
70 TaskExecutor::TaskExecutor()
71  : execute_function(execute_task)
72 {
73 }
74 
76 {
77  static TaskExecutor executor;
78  return executor;
79 }
80 } // namespace graph
81 } // namespace arm_compute
void operator()()
Function operator.
Definition: Workload.cpp:36
void prepare()
Prepare execution task.
Definition: Workload.cpp:62
Interface for CPU tensor.
Definition: ITensor.h:36
Copyright (c) 2017-2021 Arm Limited.
std::unique_ptr< arm_compute::IFunction > task
Task to execute.
Definition: Workload.h:87
void execute_task(ExecutionTask &task)
Definition: Workload.cpp:41
static TaskExecutor & get()
Task executor accessor.
Definition: Workload.cpp:75
void print(std::ostream &s, IOFormatInfo io_fmt=IOFormatInfo()) const
Print a tensor to a given stream using user defined formatting information.
virtual NodeType type() const =0
Returns node&#39;s type.
INode * node
Node bound to this workload.
Definition: Workload.h:88
std::function< decltype(execute_task)> execute_function
Function that is responsible for executing tasks.
Definition: Workload.h:63