From a7d41ccfc92f0a00ee6b4e96f217eb8f27956b00 Mon Sep 17 00:00:00 2001
From: Tai Ly <tai.ly@arm.com>
Date: Thu, 28 Sep 2023 18:45:51 +0000
Subject: Adjust to revised Stateful ops

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I86c2d39ff9e3f88dd25d5044f1d2d679e163cc1a
---
 src/TosaSerialize.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index b894383..45a1b37 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -307,7 +307,7 @@ std::string TosaSerializationOperatorBuilder::GetVariableTensorName(
     mlir::Operation *op) const {
   mlir::Attribute variable_op_name_attr = op->getAttr("name");
   std::string variable_tensor_mlir_name =
-      op->getAttr("name").cast<mlir::FlatSymbolRefAttr>().getValue().str();
+      op->getAttr("name").cast<mlir::StringAttr>().getValue().str();
 
   if (variable_tensor_flatbuffer_name_map.find(variable_tensor_mlir_name) ==
       variable_tensor_flatbuffer_name_map.end()) {
-- 
cgit v1.2.1