[NFC] simplify get/cast -> getAs

hunterzju · Oct 27, 2021 · 5f1d4ae · 5f1d4ae
1 parent 6628939
commit 5f1d4ae
Showing 1 changed file with 17 additions and 19 deletions.
diff --git a/lib/Conversion/FIRRTLToHW/LowerToHW.cpp b/lib/Conversion/FIRRTLToHW/LowerToHW.cpp
@@ -186,25 +186,23 @@ static void moveVerifAnno(ModuleOp top, AnnotationSet &annos,
   auto ctx = top.getContext();
   if (!anno)
     return;
-  if (auto _dir = anno.get("directory"))
-    if (auto dir = _dir.cast<StringAttr>()) {
-      SmallVector<NamedAttribute> old;
-      for (auto i : top->getAttrs())
-        old.push_back(i);
-      old.emplace_back(Identifier::get(attrBase, ctx),
-                       hw::OutputFileAttr::getAsDirectory(ctx, dir.getValue()));
-      top->setAttrs(old);
-    }
-  if (auto _file = anno.get("filename"))
-    if (auto file = _file.cast<StringAttr>()) {
-      SmallVector<NamedAttribute> old;
-      for (auto i : top->getAttrs())
-        old.push_back(i);
-      old.emplace_back(Identifier::get(attrBase + ".bindfile", ctx),
-                       hw::OutputFileAttr::getFromFilename(
-                           ctx, file.getValue(), /*excludeFromFileList=*/true));
-      top->setAttrs(old);
-    }
+  if (auto dir = anno.getAs<StringAttr>("directory")) {
+    SmallVector<NamedAttribute> old;
+    for (auto i : top->getAttrs())
+      old.push_back(i);
+    old.emplace_back(Identifier::get(attrBase, ctx),
+                     hw::OutputFileAttr::getAsDirectory(ctx, dir.getValue()));
+    top->setAttrs(old);
+  }
+  if (auto file = anno.getAs<StringAttr>("filename")) {
+    SmallVector<NamedAttribute> old;
+    for (auto i : top->getAttrs())
+      old.push_back(i);
+    old.emplace_back(Identifier::get(attrBase + ".bindfile", ctx),
+                     hw::OutputFileAttr::getFromFilename(
+                         ctx, file.getValue(), /*excludeFromFileList=*/true));
+    top->setAttrs(old);
+  }
 }
 
 static SmallVector<FirMemory> collectFIRRTLMemories(FModuleOp module) {