diff options
author | Pablo Tello <pablo.tello@arm.com> | 2021-06-09 11:50:01 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-06-09 11:03:00 +0000 |
commit | a5c428a5428d1c7a9d1d03fd198d6a8578b6c12c (patch) | |
tree | 9a68e75c2058e4b1ce92e3f59966e201fd754c48 /src | |
parent | 84f8aa7ea4dd1faf3302f6f376668c4be55d68ad (diff) | |
download | ComputeLibrary-a5c428a5428d1c7a9d1d03fd198d6a8578b6c12c.tar.gz |
Fixed segfault in NEGEMMConv2d
* Check if biases are not null before dereference
* Resolves COMPMID-4547
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Change-Id: I9b266deb849f40bd6a1510c444e2c522a03d934e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5784
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMConv2d.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMConv2d.cpp b/src/runtime/NEON/functions/NEGEMMConv2d.cpp index 94ceb6d27c..564ce2f514 100644 --- a/src/runtime/NEON/functions/NEGEMMConv2d.cpp +++ b/src/runtime/NEON/functions/NEGEMMConv2d.cpp @@ -55,7 +55,7 @@ void NEGEMMConv2d::configure(ITensor *input, const ITensor *weights, const ITens _impl->tensors.add_const_tensor(TensorType::ACL_SRC_2, biases); _impl->tensors.add_tensor(TensorType::ACL_DST, output); - _impl->op->configure(input->info(), weights->info(), biases->info(), output->info(), info); + _impl->op->configure(input->info(), weights->info(), biases != nullptr ? biases->info() : nullptr, output->info(), info); } Status NEGEMMConv2d::validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &info) |