conv:tc: Add converter for mlir(builtin) string attribute.

trailofbits · Dec 15, 2023 · 0ffc181 · 0ffc181
1 parent 8805b6a
commit 0ffc181
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/include/vast/Conversion/TypeConverters/TypeConverter.hpp b/include/vast/Conversion/TypeConverters/TypeConverter.hpp
@@ -167,4 +167,14 @@ namespace vast::conv::tc {
                 .template take_wrapped< maybe_attr_t >();
         };
     }
+
+    auto convert_string_attr(auto &type_converter) {
+        return [&type_converter](mlir::StringAttr attr) -> maybe_attr_t {
+            if (auto t = type_converter.convertType(attr.getType())) {
+                return mlir::StringAttr::get(attr.getValue(), t);
+            } else {
+                return {};
+            }
+        };
+    }
 } // namespace vast::conv::tc
diff --git a/include/vast/Conversion/TypeConverters/TypeConvertingPattern.hpp b/include/vast/Conversion/TypeConverters/TypeConvertingPattern.hpp
@@ -10,7 +10,9 @@ VAST_UNRELAX_WARNINGS
 
 #include "vast/Conversion/Common/Patterns.hpp"
 #include "vast/Conversion/TypeConverters/DataLayout.hpp"
+#include "vast/Conversion/TypeConverters/TypeConverter.hpp"
 #include "vast/Dialect/HighLevel/HighLevelDialect.hpp"
+#include "vast/Dialect/HighLevel/HighLevelOps.hpp"
 
 #include "vast/Util/Common.hpp"
 
@@ -57,6 +59,7 @@ namespace vast::conv::tc {
                 mlir::AttrTypeReplacer replacer;
                 replacer.addReplacement(conv::tc::convert_type_attr(tc));
                 replacer.addReplacement(conv::tc::convert_data_layout_attrs(tc));
+                replacer.addReplacement(conv::tc::convert_string_attr(tc));
 
                 replacer.addReplacement([&](mlir_type t) { return tc.convert_type_to_type(t); }
                 );