diff options
Diffstat (limited to 'arm_compute/graph/IDeviceBackend.h')
-rw-r--r-- | arm_compute/graph/IDeviceBackend.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/arm_compute/graph/IDeviceBackend.h b/arm_compute/graph/IDeviceBackend.h index 3a0a7e634e..8ae92e3177 100644 --- a/arm_compute/graph/IDeviceBackend.h +++ b/arm_compute/graph/IDeviceBackend.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 Arm Limited. + * Copyright (c) 2018-2019,2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -88,7 +88,8 @@ public: * * @return Backend sub-tensor handle */ - virtual std::unique_ptr<ITensorHandle> create_subtensor(ITensorHandle *parent, TensorShape shape, Coordinates coords, bool extend_parent) = 0; + virtual std::unique_ptr<ITensorHandle> + create_subtensor(ITensorHandle *parent, TensorShape shape, Coordinates coords, bool extend_parent) = 0; /** Configure a backend Node * * @note This creates an appropriate configured backend function for the given node @@ -118,6 +119,8 @@ public: * @return Weights manager */ virtual std::shared_ptr<arm_compute::IWeightsManager> create_weights_manager() = 0; + /** Synchronize kernels execution on the backend. On GPU, this results in a blocking call waiting for all kernels to be completed. */ + virtual void sync() = 0; }; } // namespace backends } // namespace graph |