COMPMID-3887: NEGEMMConvolutionLayer hangs up on num_threads>18
authorSheri Zhang <sheri.zhang@arm.com>
Thu, 5 Nov 2020 10:43:37 +0000 (10:43 +0000)
committerGeorgios Pinitas <georgios.pinitas@arm.com>
Tue, 10 Nov 2020 10:44:47 +0000 (10:44 +0000)
commitd80792a2d342439d5577b9bc0c9410fbeb70ea5d
treef42a83af63ce945a24deb5ebe0911da874c69963
parent030762611478edfb3c371da6b2249b415178ccc1
COMPMID-3887: NEGEMMConvolutionLayer hangs up on num_threads>18

When store the Tensor shape information, the size was clamped to 32bit, which cause the memory allocated is not big enough.

Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I9f7dfcd5595a143b0ed4f6973e20bcd9d776b673
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4331
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
arm_compute/core/TensorShape.h
arm_compute/core/Window.inl
utils/GraphUtils.cpp