From 5412590a0e4ddb731c04b1c6fe03690e28ef3e27 Mon Sep 17 00:00:00 2001
From: Dmitriy Smirnov <dmitriy.smirnov@arm.com>
Date: Fri, 6 Jan 2023 19:22:08 +0000
Subject: Added support of unranked tensors

Added support of unranked tensors
to TosaSerializationBlockBuilder::BuildTosaSerializationTensor
method

Signed-off-by: Dmitriy Smirnov <dmitriy.smirnov@arm.com>
Change-Id: Ia691437b7571f01a4b964a286c6d844eed80af75
---
 src/TosaSerialize.cpp | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index 547da8c..e69fcba 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1559,17 +1559,18 @@ TosaSerializationBlockBuilder::BuildTosaSerializationTensor(
     return nullptr;
   }
 
-  mlir::RankedTensorType tensor =
-      val.getType().dyn_cast<mlir::RankedTensorType>();
-  if (!tensor) {
-    llvm::errs() << "TOSA serialization, attempt to build an "
-                    "non-RankedTensorType Tensor\n";
+  auto ttype = val.getType().dyn_cast<mlir::TensorType>();
+  if (!ttype) {
+    llvm::errs() << "TOSA serialization, supplied value is not of TensorType\n";
     return nullptr;
   }
-  std::vector<int32_t> shape(tensor.getShape().begin(),
-                             tensor.getShape().end());
-  DType type = Type2DType(tensor.getElementType());
 
+  auto ranked = val.getType().dyn_cast<mlir::RankedTensorType>();
+  std::vector<int32_t> shape =
+      ttype.hasRank() ? std::vector<int32_t>(ranked.getShape().begin(),
+                                             ranked.getShape().end())
+                      : std::vector<int32_t>();
+  DType type = Type2DType(ttype.getElementType());
   ts = new TosaSerializationTensor(name, shape, type, std::vector<uint8_t>());
 
   return ts;
-- 
cgit v1.2.1