hl: Add asm label attribute.

MATRIXKOO · Nov 1, 2023 · f93e272 · f93e272
1 parent f859614
commit f93e272
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/include/vast/CodeGen/CodeGenAttrVisitor.hpp b/include/vast/CodeGen/CodeGenAttrVisitor.hpp
@@ -96,6 +96,10 @@ namespace vast::cg {
             return make< hl::BuiltinAttr >(attr->getID());
         }
 
+        mlir_attr VisitAsmLabelAttr(const clang::AsmLabelAttr *attr) {
+            return make< hl::AsmLabelAttr >(attr->getLabel(), attr->getIsLiteralLabel());
+        }
+
         mlir_attr VisitAllocAlignAttr(const clang::AllocAlignAttr *attr) {
             return make< hl::AllocAlignAttr >(attr->getParamIndex().getSourceIndex());
         }

diff --git a/include/vast/Dialect/HighLevel/HighLevelAttributes.td b/include/vast/Dialect/HighLevel/HighLevelAttributes.td
@@ -43,6 +43,20 @@ def RestrictAttr : HighLevel_Attr< "Restrict", "restrict" >;
 def NoThrowAttr  : HighLevel_Attr< "NoThrow", "nothrow" >;
 def NonNullAttr  : HighLevel_Attr< "NonNull", "nonnull" >;
 
+def AsmLabelAttr : HighLevel_Attr< "AsmLabel", "asm" > {
+  let parameters = (ins "::mlir::StringAttr":$label, "bool":$isLiteral);
+  let builders = [
+    AttrBuilderWithInferredContext<(ins "::mlir::StringAttr":$label, "bool":$literal), [{
+      return get(label.getContext(), label, literal);
+    }]>,
+    AttrBuilder<(ins "::mlir::StringRef":$label, "bool":$literal), [{
+      return get(mlir::StringAttr::get($_ctxt, label), literal);
+    }]>,
+  ];
+
+  let assemblyFormat = "`<` $label `,` $isLiteral `>`";
+}
+
 def ModeAttr : HighLevel_Attr< "Mode", "mode" > {
   let parameters = (ins "::mlir::StringAttr":$mode);
   let builders = [