[mlir][nvvm] Add populate function (nfc)

This work adds populate function for the nvvm to llvm conversion pattern.

Reviewed By: kuhar

Differential Revision: https://reviews.llvm.org/D155189

diff --git a/mlir/include/mlir/Conversion/NVVMToLLVM/NVVMToLLVM.h b/mlir/include/mlir/Conversion/NVVMToLLVM/NVVMToLLVM.h
index 00c33dfd77654878cc58e02e9d3ca21cf3102a48..383ef61f21d705ab90c8e172469512c2723dffd9 100644 (file)
--- a/mlir/include/mlir/Conversion/NVVMToLLVM/NVVMToLLVM.h
+++ b/mlir/include/mlir/Conversion/NVVMToLLVM/NVVMToLLVM.h
@@ -19,6 +19,8 @@ class Pass;
 #define GEN_PASS_DECL_CONVERTNVVMTOLLVMPASS
 #include "mlir/Conversion/Passes.h.inc"
 
+void populateNVVMToLLVMConversionPatterns(RewritePatternSet &patterns);
+
 } // namespace mlir
 
 #endif // MLIR_CONVERSION_NVVMTOLLVM_NVVMTOLLVMPASS_H_

diff --git a/mlir/lib/Conversion/NVVMToLLVM/NVVMToLLVM.cpp b/mlir/lib/Conversion/NVVMToLLVM/NVVMToLLVM.cpp
index 36c2f3ab2cfb196b56d332eb02ebd047c9586cb2..330ea77c4012956a8db82daba80fbddbcdc720bb 100644 (file)
--- a/mlir/lib/Conversion/NVVMToLLVM/NVVMToLLVM.cpp
+++ b/mlir/lib/Conversion/NVVMToLLVM/NVVMToLLVM.cpp
@@ -183,7 +183,7 @@ struct ConvertNVVMToLLVMPass
     ConversionTarget target(getContext());
     target.addLegalDialect<::mlir::LLVM::LLVMDialect>();
     RewritePatternSet pattern(&getContext());
-    pattern.add<PtxLowering>(pattern.getContext());
+    mlir::populateNVVMToLLVMConversionPatterns(pattern);
     if (failed(
             applyPartialConversion(getOperation(), target, std::move(pattern))))
       signalPassFailure();
@@ -191,3 +191,7 @@ struct ConvertNVVMToLLVMPass
 };
 
 } // namespace
+
+void mlir::populateNVVMToLLVMConversionPatterns(RewritePatternSet &patterns) {
+  patterns.add<PtxLowering>(patterns.getContext());
+}