aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/ClWorkloadFactory.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/ClWorkloadFactory.hpp')
-rw-r--r--src/armnn/backends/ClWorkloadFactory.hpp11
1 files changed, 9 insertions, 2 deletions
diff --git a/src/armnn/backends/ClWorkloadFactory.hpp b/src/armnn/backends/ClWorkloadFactory.hpp
index 2477e23eeb..e1e66c050b 100644
--- a/src/armnn/backends/ClWorkloadFactory.hpp
+++ b/src/armnn/backends/ClWorkloadFactory.hpp
@@ -23,18 +23,22 @@ namespace armnn
{
class IClTunedParameters;
+class ClTunedParameters;
// ARM Compute OpenCL workload factory
class ClWorkloadFactory : public IWorkloadFactory
{
public:
- virtual ~ClWorkloadFactory(){};
+
+ ClWorkloadFactory(IClTunedParameters* clTunedParameters = nullptr);
+
+ virtual ~ClWorkloadFactory();
virtual Compute GetCompute() const override { return Compute::GpuAcc; }
static bool IsLayerSupported(const Layer& layer, DataType dataType, std::string& outReasonIfUnsupported);
- void LoadOpenClRuntime(IClTunedParameters* clTunedParameters = nullptr);
+ void LoadOpenClRuntime();
virtual bool SupportsSubTensors() const override { return true; }
@@ -109,6 +113,9 @@ public:
virtual std::unique_ptr<IWorkload> CreateFloor(const FloorQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;
+
+private:
+ ClTunedParameters* m_clTunedParameters;
};
class ClTunedParameters : public IClTunedParameters