aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h')
-rw-r--r--arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h b/arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h
index 326880f721..b5af589cd2 100644
--- a/arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h
+++ b/arm_compute/dynamic_fusion/runtime/gpu/cl/ClWorkloadRuntime.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2022 Arm Limited.
+ * Copyright (c) 2022-2023 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -24,8 +24,8 @@
#ifndef ARM_COMPUTE_DYNAMIC_FUSION_RUNTIME_GPU_CL_CLWORKLOADRUNTIME
#define ARM_COMPUTE_DYNAMIC_FUSION_RUNTIME_GPU_CL_CLWORKLOADRUNTIME
+#include "arm_compute/core/TensorInfo.h"
#include "arm_compute/dynamic_fusion/sketch/MemoryDescriptor.h"
-
#include <map>
#include <memory>
@@ -63,7 +63,7 @@ public:
Status run(const std::vector<CLTensor *> &tensors);
/** Get auxiliary tensors of the workload and their memory requirement
*/
- std::vector<std::pair<CLTensor *, AuxMemoryInfo>> get_auxiliary_tensors();
+ std::vector<std::tuple<CLTensor *, TensorInfo, AuxMemoryInfo>> get_auxiliary_tensors();
private:
/** Enqueue prepare workload