From 4ea9bac9d7d198b1bb06b42ded6aff9c9a653343 Mon Sep 17 00:00:00 2001 From: Adnan AlSinan Date: Fri, 13 Oct 2023 16:51:58 +0100 Subject: Fix memory Error in Reverse Fixture. Resolves: COMPMID-6581 Signed-off-by: Adnan AlSinan Change-Id: I0a634e064377e54b9190241c01fc75c212522ba7 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10481 Reviewed-by: Viet-Hoa Do Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Benchmark: Arm Jenkins --- tests/validation/NEON/Reverse.cpp | 2 +- tests/validation/fixtures/ReverseFixture.h | 10 +++++----- 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 generate_random_axis(bool use_negative = false) + std::vector generate_random_axis(bool use_negative = false) { - std::vector axis_v; + std::vector 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(axis_v.begin(), axis_v.begin() + _num_dims); + axis_v = std::vector(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(axis_data.data())); + std::copy(axis_v.begin(), axis_v.begin() + axis_shape.total_size(), static_cast(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(src, axis, use_inverted_axis); } -- cgit v1.2.1