diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-02-19 13:42:59 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-02-19 14:53:38 +0000 |
commit | f85433be70e3d8d38025835e3f3fc450a54a2598 (patch) | |
tree | b547b075f3b0fdc0091fb8242f93a7e089d57435 /src/runtime/NEON/functions/NEGEMM.cpp | |
parent | e46a7beb6b3d89b7cc6a96faeab19ee4478d4e2e (diff) | |
download | ComputeLibrary-f85433be70e3d8d38025835e3f3fc450a54a2598.tar.gz |
COMPMID-2006: NEON GEMMLowp assertion failure.
Mark auxilary tensors as resizable when cloning.
Change-Id: I582e6d09a7daadbc43cf02f46a53e51c178daacb
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/726
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMM.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMM.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEGEMM.cpp b/src/runtime/NEON/functions/NEGEMM.cpp index 99cddc2920..914f088bf5 100644 --- a/src/runtime/NEON/functions/NEGEMM.cpp +++ b/src/runtime/NEON/functions/NEGEMM.cpp @@ -91,8 +91,8 @@ void NEGEMM::configure(const ITensor *a, const ITensor *b, const ITensor *c, ITe shape_tmp_b.set(0, b->info()->dimension(1) * transpose_w); shape_tmp_b.set(1, std::ceil(b->info()->dimension(0) / static_cast<float>(transpose_w))); - TensorInfo info_a = a->info()->clone()->set_tensor_shape(shape_tmp_a); - TensorInfo info_b = b->info()->clone()->set_tensor_shape(shape_tmp_b); + TensorInfo info_a = a->info()->clone()->set_tensor_shape(shape_tmp_a).set_is_resizable(true); + TensorInfo info_b = b->info()->clone()->set_tensor_shape(shape_tmp_b).set_is_resizable(true); _tmp_a.allocator()->init(info_a); _tmp_b.allocator()->init(info_b); |