hl: Add to_hlbi to simplify pipeline.

trailofbits · Mar 12, 2024 · e325d84 · e325d84
1 parent 363fdb8
commit e325d84
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/include/vast/Conversion/Passes.hpp b/include/vast/Conversion/Passes.hpp
@@ -65,6 +65,7 @@ namespace vast
 
     namespace conv::pipeline
     {
+        pipeline_step_ptr to_hlbi();
         pipeline_step_ptr abi();
         pipeline_step_ptr irs_to_llvm();
         pipeline_step_ptr core_to_llvm();

diff --git a/lib/vast/Conversion/FromHL/Passes.cpp b/lib/vast/Conversion/FromHL/Passes.cpp
@@ -11,6 +11,10 @@ VAST_UNRELAX_WARNINGS
 
 namespace vast::conv::pipeline {
 
+    pipeline_step_ptr to_hlbi() {
+        return pass(createHLToHLBI);
+    }
+
     pipeline_step_ptr hl_to_ll_cf() {
         // TODO add dependencies
         return pass(createHLToLLCFPass);

diff --git a/lib/vast/Dialect/HighLevel/Passes.cpp b/lib/vast/Dialect/HighLevel/Passes.cpp
@@ -7,6 +7,7 @@ VAST_RELAX_WARNINGS
 #include <mlir/Pass/PassManager.h>
 VAST_UNRELAX_WARNINGS
 
+#include "vast/Conversion/Passes.hpp"
 #include "vast/Dialect/HighLevel/Passes.hpp"
 
 namespace vast::hl::pipeline {
@@ -52,7 +53,7 @@ namespace vast::hl::pipeline {
     }
 
     pipeline_step_ptr simplify() {
-        return compose("simplify", ude, dce, desugar);
+        return compose("simplify", conv::pipeline::to_hlbi, ude, dce, desugar);
     }
 
     //