commit 2098445ecfcb8379037f8f287e98a897569419b9
author Freddie Liardet <frederick.liardet@arm.com> Wed May 12 12:09:12 2021 +0100
committer frederick.liardet <frederick.liardet@arm.com> Wed Jun 02 10:28:36 2021 +0000
tree b99886ad2706e5c7471793b992897772550c507a
parent 760b494acdd6c0184e15b86e40256b27d6312b7a

Fix bug in PReluLayer when input is 1xN size

Fix issue where gpu elementwise operation kernel would not compile when
input is 1xN size and prelu is the chosen operator.

Add relevant tests for 1xN input.

Signed-off-by: Freddie Liardet <frederick.liardet@arm.com>
Change-Id: If0651cfa399ca1d9c65f2632b75536c7931f27d4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5760
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

tests/validation/CL/PReluLayer.cpp

diff --git a/tests/validation/CL/PReluLayer.cpp b/tests/validation/CL/PReluLayer.cpp
index 82f3e4f..043262d 100644
--- a/tests/validation/CL/PReluLayer.cpp
+++ b/tests/validation/CL/PReluLayer.cpp
@@ -200,6 +200,11 @@
     // Validate output
     validate(CLAccessor(_target), _reference);
 }
+FIXTURE_DATA_TEST_CASE(RunOneDimensional, CLPReluLayerFixture<int16_t>, framework::DatasetMode::ALL, combine(framework::dataset::make("Shape", TensorShape(1U, 16U)), PReluLayerS16Dataset))
+{
+    // Validate output
+    validate(CLAccessor(_target), _reference);
+}
 TEST_SUITE_END()
 
 TEST_SUITE(Float)