diff options
| -rw-r--r-- | serialization/tosa_serialization_handler.cpp | 10 |
1 files 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); } |