[Mlir-commits] [mlir] [mlir][spirv] Fix serialization of TensorARM with rank higher than one (PR #152391)

[Mohammadreza Ameri Mahabadian]

================
@@ -956,6 +956,11 @@ Serializer::prepareDenseElementsConstant(Location loc, Type constType,
   uint32_t resultID = getNextID();
   SmallVector<uint32_t, 4> operands = {typeID, resultID};
   auto elementType = cast<spirv::CompositeType>(constType).getElementType(0);
+  if (auto tensorArmType = dyn_cast<spirv::TensorArmType>(constType)) {
+    ArrayRef<int64_t> innerShape = tensorArmType.getShape().drop_front();
+    if (innerShape.size() > 0)
----------------
mahabadm wrote:

Done.

https://github.com/llvm/llvm-project/pull/152391