Add a utility function to populate StdOp to SPIRV Conversion Patterns

The function populateStdOpsToSPIRVPatterns appends the conversion
patterns automatically generated from StdOpsToSPIRVConversion.td to a
list of patterns

PiperOrigin-RevId: 259677890

diff --git a/mlir/include/mlir/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.h b/mlir/include/mlir/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.h
new file mode 100644 (file)
index 0000000..7e75430
--- /dev/null
+++ b/mlir/include/mlir/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.h
@@ -0,0 +1,35 @@
+//===- StdOpsToSPIRVConversion.h - Convert StandardOps to SPIR-V *- C++ -*-===//
+//
+// Copyright 2019 The MLIR Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// =============================================================================
+//
+// This file defines utility function to import patterns to convert StandardOps
+// to SPIR-V ops
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef STANDARD_OPS_TO_SPIRV_H_
+#define STANDARD_OPS_TO_SPIRV_H_
+
+#include "mlir/IR/PatternMatch.h"
+
+namespace mlir {
+/// Method to append to a pattern list additional patterns for translating
+/// StandardOps to SPIR-V ops.
+void populateStdOpsToSPIRVPatterns(MLIRContext *context,
+                                   OwningRewritePatternList &patterns);
+} // namespace mlir
+
+#endif // STANDARD_OPS_TO_SPIRV_H_

diff --git a/mlir/lib/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.cpp b/mlir/lib/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.cpp
index 6ba4c8b..45213bb 100644 (file)
--- a/mlir/lib/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.cpp
+++ b/mlir/lib/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.cpp
@@ -20,10 +20,10 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include "mlir/Conversion/StandardToSPIRV/StdOpsToSPIRVConversion.h"
 #include "mlir/Dialect/SPIRV/Passes.h"
 #include "mlir/Dialect/SPIRV/SPIRVOps.h"
 #include "mlir/IR/Operation.h"
-#include "mlir/IR/PatternMatch.h"
 #include "mlir/IR/StandardTypes.h"
 #include "mlir/StandardOps/Ops.h"
 
@@ -39,11 +39,18 @@ class StdOpsToSPIRVConversionPass
 #include "StdOpsToSPIRVConversion.cpp.inc"
 } // namespace
 
+namespace mlir {
+void populateStdOpsToSPIRVPatterns(MLIRContext *context,
+                                   OwningRewritePatternList &patterns) {
+  populateWithGenerated(context, &patterns);
+}
+} // namespace mlir
+
 void StdOpsToSPIRVConversionPass::runOnFunction() {
   OwningRewritePatternList patterns;
   auto func = getFunction();
 
-  populateWithGenerated(func.getContext(), &patterns);
+  populateStdOpsToSPIRVPatterns(func.getContext(), patterns);
   applyPatternsGreedily(func, std::move(patterns));
 }