diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-02-14 14:18:01 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
commit | e3fba0afa892c66379da1e3d3843f2155a1fb29a (patch) | |
tree | 2a420a3d988d269cc157d73e8d3c8accf1d21af8 /utils/GraphUtils.h | |
parent | f07d28d9ee8ae73a93fe433f72855b6dcf58ad90 (diff) | |
download | ComputeLibrary-e3fba0afa892c66379da1e3d3843f2155a1fb29a.tar.gz |
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 <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'utils/GraphUtils.h')
-rw-r--r-- | utils/GraphUtils.h | 6 |
1 files changed, 3 insertions, 3 deletions
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<graph::ITensorAccessor> 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; |