Make checkConstantOutputs and populateAffineAndKrnlToLLVMConversion s…

…harable for other drivers (llvm#1213) Signed-off-by: Tung D. Le <tung@jp.ibm.com>
nod-ai · Mar 3, 2022 · 77fac29 · 77fac29
1 parent fd3e930
commit 77fac29
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/Conversion/KrnlToLLVM/ConvertKrnlToLLVM.cpp b/src/Conversion/KrnlToLLVM/ConvertKrnlToLLVM.cpp
@@ -61,7 +61,7 @@ using namespace mlir;
 namespace onnx_mlir {
 namespace krnl {
 
-static void checkConstantOutputs(
+void checkConstantOutputs(
     ModuleOp &module, SmallVectorImpl<bool> &constantOutputs) {
   Operation *entryPointOp;
   auto walkResult = module->walk([&](mlir::Operation *op) -> WalkResult {
@@ -136,7 +136,7 @@ static void checkConstantOutputs(
   }
 }
 
-static void populateAffineAndKrnlToLLVMConversion(RewritePatternSet &patterns,
+void populateAffineAndKrnlToLLVMConversion(RewritePatternSet &patterns,
     LLVMTypeConverter &typeConverter, MLIRContext *ctx,
     ArrayRef<bool> constantOutputs, bool singleEntryPoint) {
   // TODO: look at what is done in

diff --git a/src/Conversion/KrnlToLLVM/ConvertKrnlToLLVM.hpp b/src/Conversion/KrnlToLLVM/ConvertKrnlToLLVM.hpp
@@ -24,6 +24,10 @@ const std::string DEFAULT_DYN_ENTRY_POINT = "run_main_graph";
 namespace onnx_mlir {
 namespace krnl {
 
+void populateAffineAndKrnlToLLVMConversion(RewritePatternSet &patterns,
+    LLVMTypeConverter &typeConverter, MLIRContext *ctx,
+    ArrayRef<bool> constantOutputs, bool singleEntryPoint);
+
 void populateKrnlToLLVMConversion(mlir::LLVMTypeConverter &typeConverter,
     mlir::RewritePatternSet &patterns, mlir::MLIRContext *ctx,
     llvm::ArrayRef<bool> constantOutputs, bool singleEntryPoint);
@@ -71,6 +75,9 @@ void populateLoweringKrnlVectorTypeCastOpPattern(
     LLVMTypeConverter &typeConverter, mlir::RewritePatternSet &patterns,
     mlir::MLIRContext *ctx);
 
+void checkConstantOutputs(
+    ModuleOp &module, SmallVectorImpl<bool> &constantOutputs);
+
 void recordEntryPointSignatures(mlir::ModuleOp &module,
     llvm::SmallVectorImpl<std::string> &entryPointNames,
     llvm::SmallVectorImpl<std::string> &inSignatures,