diff options
author | Adnan AlSinan <adnan.alsinan@arm.com> | 2023-10-13 16:51:58 +0100 |
---|---|---|
committer | Adnan AlSinan <adnan.alsinan@arm.com> | 2023-10-17 12:59:23 +0000 |
commit | 4ea9bac9d7d198b1bb06b42ded6aff9c9a653343 (patch) | |
tree | 116f54c7a9ca90fef4f3313599eb858a512c52b9 /tests/validation/fixtures/ReverseFixture.h | |
parent | 098efc43bb0c7a3585f9d615e35e43c0dbf9dcfa (diff) | |
download | ComputeLibrary-4ea9bac9d7d198b1bb06b42ded6aff9c9a653343.tar.gz |
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>
Diffstat (limited to 'tests/validation/fixtures/ReverseFixture.h')
-rw-r--r-- | tests/validation/fixtures/ReverseFixture.h | 10 |
1 files changed, 5 insertions, 5 deletions
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); } |