From e3fba0afa892c66379da1e3d3843f2155a1fb29a Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 14 Feb 2018 14:18:01 +0000 Subject: COMPMID-925: Enabling OpenCL tuner in the graph examples Change-Id: I4fe501281f527e20e8fdd0253d59ea2c4629056b Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120354 Tested-by: Jenkins Reviewed-by: Anthony Barbier Reviewed-by: Georgios Pinitas --- utils/GraphUtils.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'utils') diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index da52c26520..b48e5b563d 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -256,14 +256,14 @@ inline std::unique_ptr get_input_accessor(const std::str /** Utility function to return the TargetHint * - * @param[in] target Integer value which expresses the selected target. Must be 0 for NEON or 1 for OpenCL + * @param[in] target Integer value which expresses the selected target. Must be 0 for NEON, 1 for OpenCL or 2 for OpenCL with Tuner * * @return the TargetHint */ inline graph::TargetHint set_target_hint(int target) { - ARM_COMPUTE_ERROR_ON_MSG(target > 1, "Invalid target. Target must be 0 (NEON) or 1 (OpenCL)"); - if(target == 1 && graph::Graph::opencl_is_available()) + ARM_COMPUTE_ERROR_ON_MSG(target > 2, "Invalid target. Target must be 0 (NEON), 1 (OpenCL) or 2 (OpenCL with Tuner)"); + if((target == 1 || target == 2) && graph::Graph::opencl_is_available()) { // If type of target is OpenCL, check if OpenCL is available and initialize the scheduler return graph::TargetHint::OPENCL; -- cgit v1.2.1