Fix memory Error in Reverse Fixture.

Resolves: COMPMID-6581

Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: I0a634e064377e54b9190241c01fc75c212522ba7
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10481
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>

2 files changed, 6 insertions, 6 deletions

diff --git a/tests/validation/NEON/Reverse.cpp b/tests/validation/NEON/Reverse.cpp
index eb0c995ad9..7b5337f14b 100644
--- a/tests/validation/NEON/Reverse.cpp
+++ b/tests/validation/NEON/Reverse.cpp
@@ -44,7 +44,7 @@ namespace validation
 namespace
 {
 using framework::dataset::make;
-auto run_small_dataset = combine(datasets::SmallShapes(), datasets::Tiny1DShapes());
+auto run_small_dataset = combine(datasets::Small3DShapes(), datasets::Tiny1DShapes());
 auto run_large_dataset = combine(datasets::LargeShapes(), datasets::Tiny1DShapes());
 
 } // namespace
diff --git a/tests/validation/fixtures/ReverseFixture.h b/tests/validation/fixtures/ReverseFixture.h
index 3bf4c3e327..d53945191e 100644
--- a/tests/validation/fixtures/ReverseFixture.h
+++ b/tests/validation/fixtures/ReverseFixture.h
@@ -86,9 +86,9 @@ protected:
     {
         library->fill_tensor_uniform(tensor, 0);
     }
-    std::vector<int> generate_random_axis(bool use_negative = false)
+    std::vector<int32_t> generate_random_axis(bool use_negative = false)
     {
-        std::vector<int> axis_v;
+        std::vector<int32_t> axis_v;
         if(use_negative)
         {
             axis_v = { -1, -2, -3, -4 };
@@ -97,7 +97,7 @@ protected:
         {
             axis_v = { 0, 1, 2, 3 };
         }
-        axis_v = std::vector<int>(axis_v.begin(), axis_v.begin() + _num_dims);
+        axis_v = std::vector<int32_t>(axis_v.begin(), axis_v.begin() + _num_dims);
         std::mt19937 g(library->seed());
         std::shuffle(axis_v.begin(), axis_v.end(), g);
 
@@ -134,7 +134,7 @@ protected:
         {
             auto axis_data = AccessorType(axis);
             auto axis_v    = generate_random_axis(use_negative_axis);
-            std::copy(axis_v.begin(), axis_v.begin() + _num_dims, static_cast<int32_t *>(axis_data.data()));
+            std::copy(axis_v.begin(), axis_v.begin() + axis_shape.total_size(), static_cast<int32_t *>(axis_data.data()));
         }
 
         // Compute function
@@ -152,7 +152,7 @@ protected:
         // Fill reference
         fill(src);
         auto axis_v = generate_random_axis(use_negative_axis);
-        std::copy(axis_v.begin(), axis_v.begin() + _num_dims, axis.data());
+        std::copy(axis_v.begin(), axis_v.begin() + axis_shape.total_size(), axis.data());
 
         return reference::reverse<T>(src, axis, use_inverted_axis);
     }