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>
diff --git a/src/runtime/NEON/functions/NEGEMMConv2d.cpp b/src/runtime/NEON/functions/NEGEMMConv2d.cpp
index 94ceb6d..564ce2f 100644
--- a/src/runtime/NEON/functions/NEGEMMConv2d.cpp
+++ b/src/runtime/NEON/functions/NEGEMMConv2d.cpp
@@ -55,7 +55,7 @@
_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)