diff options
| author | Orlaith Monahan <david.monahan@arm.com> | 2024-01-23 13:52:30 +0000 |
|---|---|---|
| committer | David Monahan <david.monahan@arm.com> | 2024-01-23 16:06:24 +0000 |
| commit | e1ac86975cb8e19a76c1389073147ba306573b3f (patch) | |
| tree | bfdd72173073a4623d4f9f2a8031912b71ce5ec9 /src/backends/gpuFsa/GpuFsaBackend.hpp | |
| parent | 91ffe3d84528a68a68de8f8a1a36a1a2177233a5 (diff) | |
| download | armnn-e1ac86975cb8e19a76c1389073147ba306573b3f.tar.gz | |
IVGCVSW-7628 Update GpuFsa to use ACLs latest fixes
* Updates to the existing GpuFsa backend to incorporate patch:
https://review.mlplatform.org/c/ml/ComputeLibrary/+/10990
* Update the ACL pin to include the patch with the fixes
Change-Id: I08d111265f4617657ee7f20249aeb111f64ba7a9
Signed-off-by: David Monahan <david.monahan@arm.com>
Diffstat (limited to 'src/backends/gpuFsa/GpuFsaBackend.hpp')
| -rw-r--r-- | src/backends/gpuFsa/GpuFsaBackend.hpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/backends/gpuFsa/GpuFsaBackend.hpp b/src/backends/gpuFsa/GpuFsaBackend.hpp index 4c2a5f02e3..f5a866b186 100644 --- a/src/backends/gpuFsa/GpuFsaBackend.hpp +++ b/src/backends/gpuFsa/GpuFsaBackend.hpp @@ -23,10 +23,10 @@ namespace armnn /** * A structure which contains all the elements needed to execute a fused workload in the GpuFsa Backend * - * @param[in, out] sketch A unique pointer to the sketch containing the operators which have been fused. - * @param[in, out] TensorInfos A shared pointer to a GpuWorkloadContext which contains TensorInfos - * @param[in, out] inputIds A unique pointer to a vector of input Ids used to access workloadContext TensorInfos - * @param[in, out] outputIds A unique pointer to a vector of output Ids used to access workloadContext TensorInfos + * @param[in, out] sketch A unique pointer to the sketch containing the operators which have been fused. + * @param[in, out] TensorInfos A shared pointer to a GpuWorkloadContext which creates + stores TensorInfos + * @param[in, out] inputTensorInfos A unique pointer to a vector of inputTensorInfos used by the sketch + * @param[in, out] outputTensorInfos A unique pointer to a vector of outputTensorInfos used by the sketch * */ struct GpuFsaPreCompiledBlob @@ -34,8 +34,8 @@ struct GpuFsaPreCompiledBlob std::unique_ptr<arm_compute::experimental::dynamic_fusion::GpuWorkloadSketch> sketch = nullptr; std::shared_ptr<arm_compute::experimental::dynamic_fusion::GpuWorkloadContext> workloadContext = nullptr; - std::unique_ptr<std::vector<int32_t>> inputIds = nullptr; - std::unique_ptr<std::vector<int32_t>> outputIds = nullptr; + std::unique_ptr<std::vector<arm_compute::ITensorInfo*>> inputTensorInfos = nullptr; + std::unique_ptr<std::vector<arm_compute::ITensorInfo*>> outputTensorInfos = nullptr; }; // add new capabilities here.. |