From 80b371d6db746c8dcb093c8d3fcca775403759d6 Mon Sep 17 00:00:00 2001
From: Muxianesty <myulitvinov@edu.hse.ru>
Date: Wed, 14 Aug 2024 17:24:06 +0300
Subject: [PATCH] Fixed OffsetOp printing.

---
 src/model/dfcir/lib/dfcir/DFCIROperations.cpp | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/src/model/dfcir/lib/dfcir/DFCIROperations.cpp b/src/model/dfcir/lib/dfcir/DFCIROperations.cpp
index 70f60a0..93938ce 100644
--- a/src/model/dfcir/lib/dfcir/DFCIROperations.cpp
+++ b/src/model/dfcir/lib/dfcir/DFCIROperations.cpp
@@ -419,12 +419,7 @@ void mlir::dfcir::OffsetOp::print(OpAsmPrinter &printer) {
   ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
   elidedAttrs.push_back("offset");
   printer.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
-  printer << " : ";
-  auto type = getRes().getType();
-  if (auto validType = llvm::dyn_cast<DFCIRStreamType>(type)) {
-    printer.printStrippedAttrOrType(validType);
-  } else
-    printer << type;
+  printer << " : " << getRes().getType();
 }
 
 } // namespace mlir::dfcir