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
This commit is contained in:
Mahesh Ravishankar 2019-07-23 22:38:23 -07:00 committed by A. Unique TensorFlower
parent e8bd81ba1a
commit 2ad92b6e50
2 changed files with 44 additions and 2 deletions

View File

@ -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_

View File

@ -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));
}