cg: Add enableif/diagnoseif attrs.

trailofbits · Nov 4, 2024 · 7437fcd · 7437fcd
1 parent 513c6e8
commit 7437fcd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/include/vast/CodeGen/DefaultAttrVisitor.hpp b/include/vast/CodeGen/DefaultAttrVisitor.hpp
@@ -49,11 +49,13 @@ namespace vast::cg {
         mlir_attr VisitAsmLabelAttr(const clang::AsmLabelAttr *attr);
         mlir_attr VisitAllocAlignAttr(const clang::AllocAlignAttr *attr);
         mlir_attr VisitAllocSizeAttr(const clang::AllocSizeAttr *attr);
+        mlir_attr VisitEnableIfAttr(const clang::EnableIfAttr *attr);
         mlir_attr VisitLeafAttr(const clang::LeafAttr *attr);
         mlir_attr VisitColdAttr(const clang::ColdAttr *attr);
         mlir_attr VisitDeprecatedAttr(const clang::DeprecatedAttr *attr);
         mlir_attr VisitTransparentUnionAttr(const clang::TransparentUnionAttr *attr);
         mlir_attr VisitUnavailableAttr(const clang::UnavailableAttr *attr);
+        mlir_attr VisitDiagnoseIfAttr(const clang::DiagnoseIfAttr *attr);
         mlir_attr VisitReturnsTwiceAttr(const clang::ReturnsTwiceAttr *attr);
         mlir_attr VisitMayAliasAttr(const clang::MayAliasAttr *attr);
         mlir_attr VisitUnusedAttr(const clang::UnusedAttr *attr);

diff --git a/lib/vast/CodeGen/DefaultAttrVisitor.cpp b/lib/vast/CodeGen/DefaultAttrVisitor.cpp
@@ -109,6 +109,10 @@ namespace vast::cg
         return make< hl::AllocSizeAttr >(attr->getElemSizeParam().getSourceIndex(), num);
     }
 
+    mlir_attr default_attr_visitor::VisitEnableIfAttr(const clang::EnableIfAttr *attr) {
+        return make< hl::EnableIfAttr >();
+    }
+
     mlir_attr default_attr_visitor::VisitLeafAttr(const clang::LeafAttr *attr) {
         return make< hl::LeafAttr >();
     }
@@ -129,6 +133,10 @@ namespace vast::cg
         return make< hl::UnavailableAttr >(attr->getMessage());
     }
 
+    mlir_attr default_attr_visitor::VisitDiagnoseIfAttr(const clang::DiagnoseIfAttr *attr) {
+        return make< hl::DiagnoseIfAttr >();
+    }
+
     mlir_attr default_attr_visitor::VisitReturnsTwiceAttr(const clang::ReturnsTwiceAttr *attr) {
         return make< hl::ReturnsTwiceAttr >();
     }