From a53a7bb46cdd5cdb6ef14703e0be00555c710a45 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Mon, 10 Sep 2018 18:22:03 +0100 Subject: [MLCE-54] graph_alexnet example throw an exception The 32bit version of AlexNet seems to require ~1.4 GB of memory to run (got this result via valgrind's massif tool). The problem is a miscalculation of the shape of the _transformed_b tensor, which should not be multiplied by the size of the data type, but rather represent only the size of the dimensions. Change-Id: I55dfb538b31772afc4ce0ce0ccd31fdeb1585cfb Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147666 Reviewed-by: Georgios Pinitas Reviewed-by: Anthony Barbier Tested-by: Jenkins --- .../NEON/kernels/assembly/NEGEMMInterleavedPrepareBWrapperKernel.cpp | 1 - 1 file changed, 1 deletion(-) (limited to 'src/core/NEON/kernels/assembly') diff --git a/src/core/NEON/kernels/assembly/NEGEMMInterleavedPrepareBWrapperKernel.cpp b/src/core/NEON/kernels/assembly/NEGEMMInterleavedPrepareBWrapperKernel.cpp index f33a14f2af..6c201cedb3 100644 --- a/src/core/NEON/kernels/assembly/NEGEMMInterleavedPrepareBWrapperKernel.cpp +++ b/src/core/NEON/kernels/assembly/NEGEMMInterleavedPrepareBWrapperKernel.cpp @@ -89,7 +89,6 @@ unsigned int get_B_pretransposed_array_size(unsigned int N, unsigned int K, cons // Calculate the total size of the buffer: size_t total = num_full_k * normal_k_size * (num_full_x * normal_x_size + left_over_x_size); total += left_over_k_size * (left_over_x_size + num_full_x * normal_x_size); - total *= sizeof(To); return total; } -- cgit v1.2.1