diff options
Diffstat (limited to 'src/graph')
-rw-r--r-- | src/graph/Graph.cpp | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/src/graph/Graph.cpp b/src/graph/Graph.cpp index ac5316f55e..e14bea0846 100644 --- a/src/graph/Graph.cpp +++ b/src/graph/Graph.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -215,11 +215,25 @@ void Graph::add_tensor_object(std::unique_ptr<ITensorObject> tensor) _pimpl->_graph_output->allocate(); } } + bool Graph::opencl_is_available() { return arm_compute::opencl_is_available(); } +arm_compute::GPUTarget Graph::gpu_target() +{ + // Check if OpenCL is available before returning the GPU target + if(opencl_is_available()) + { + return arm_compute::CLScheduler::get().target(); + } + else + { + return GPUTarget::MIDGARD; + } +} + void Graph::set_temp(TensorInfo &&tmp) { ARM_COMPUTE_ERROR_ON(_pimpl->_graph_input == nullptr); |