diff options
Diffstat (limited to 'src/backends/cl/workloads/ClConstantWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClConstantWorkload.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClConstantWorkload.hpp b/src/backends/cl/workloads/ClConstantWorkload.hpp index e5a1d4410d..8fa5d632c2 100644 --- a/src/backends/cl/workloads/ClConstantWorkload.hpp +++ b/src/backends/cl/workloads/ClConstantWorkload.hpp @@ -8,6 +8,8 @@ #include <arm_compute/core/Error.h> #include <backendsCommon/Workload.hpp> +#include <arm_compute/core/CL/CLCompileContext.h> + namespace armnn { arm_compute::Status ClConstantWorkloadValidate(const TensorInfo& output); @@ -15,7 +17,9 @@ arm_compute::Status ClConstantWorkloadValidate(const TensorInfo& output); class ClConstantWorkload : public BaseWorkload<ConstantQueueDescriptor> { public: - ClConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info); + ClConstantWorkload(const ConstantQueueDescriptor& descriptor, + const WorkloadInfo& info, + const arm_compute::CLCompileContext& clCompileContext); void Execute() const override; |