commit 2ea38c7f56d57a4fdcf709c9d61b7bdfab4ebfd9
author Teresa Charlin <teresa.charlinreyes@arm.com> Tue Nov 02 16:43:57 2021 +0000
committer Colm Donelan <colm.donelan@arm.com> Wed Nov 03 14:03:03 2021 +0000
tree d422855474c36758a4616c8ce3b7ad33510acef5
parent f39f8d8597c59057118f67cacf70d246f95fea9b

Bug Fix in CpuAcc and GpuAcc workload: Gather has 2 inputs

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ifbb9332c78c843c5b937c4e1b50a8f5a75409e73

src/backends/cl/workloads/ClGatherWorkload.cpp

diff --git a/src/backends/cl/workloads/ClGatherWorkload.cpp b/src/backends/cl/workloads/ClGatherWorkload.cpp
index b2341b8..06fa5af 100644
--- a/src/backends/cl/workloads/ClGatherWorkload.cpp
+++ b/src/backends/cl/workloads/ClGatherWorkload.cpp
@@ -37,7 +37,7 @@
                                          info,
                                          this->GetGuid());
 
-    m_Data.ValidateInputsOutputs("ClGatherWorkload", 1, 1);
+    m_Data.ValidateInputsOutputs("ClGatherWorkload", 2, 1);
 
     arm_compute::ICLTensor& input    = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ICLTensor& indices  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();

src/backends/neon/workloads/NeonGatherWorkload.cpp

diff --git a/src/backends/neon/workloads/NeonGatherWorkload.cpp b/src/backends/neon/workloads/NeonGatherWorkload.cpp
index f5c8d34..4cb17ca 100644
--- a/src/backends/neon/workloads/NeonGatherWorkload.cpp
+++ b/src/backends/neon/workloads/NeonGatherWorkload.cpp
@@ -34,7 +34,7 @@
                                          info,
                                          this->GetGuid());
 
-    m_Data.ValidateInputsOutputs("NeonGatherWorkload", 1, 1);
+    m_Data.ValidateInputsOutputs("NeonGatherWorkload", 2, 1);
 
     arm_compute::ITensor& input   = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
     arm_compute::ITensor& indices = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();