From 4025315d4d09e314e46b037f8c47e2096b20ddf8 Mon Sep 17 00:00:00 2001
From: Kevin Cheng <kevin.cheng@arm.com>
Date: Wed, 30 Dec 2020 10:12:35 -0800
Subject: Numpy utility supports scalar (rank 0) .npy

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I4f53a83a0de333bd851fae4e894eb87b94902637
---
 serialization/tosa_serialization_handler.cpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/serialization/tosa_serialization_handler.cpp b/serialization/tosa_serialization_handler.cpp
index 7fe9f47..ad33d23 100644
--- a/serialization/tosa_serialization_handler.cpp
+++ b/serialization/tosa_serialization_handler.cpp
@@ -1225,8 +1225,14 @@ NumpyUtilities::NPError NumpyUtilities::checkNpyHeader(FILE* infile, const uint3
                 if (!ptr)
                     break;
 
-                shape.push_back(atoi(ptr));
-                totalElems *= atoi(ptr);
+                int dim = atoi(ptr);
+
+                // Dimension is 0
+                if (dim == 0)
+                    break;
+
+                shape.push_back(dim);
+                totalElems *= dim;
                 ptr = strtok_r(NULL, ",", &end);
             }
 
-- 
cgit v1.2.1