github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/mlir/lib/Conversion/StandardToSPIRV/ConvertStandardToSPIRVPass.cpp (about)

     1  //===- ConvertStandardToSPIRVPass.cpp - Convert Std Ops to SPIR-V Ops -----===//
     2  //
     3  // Copyright 2019 The MLIR Authors.
     4  //
     5  // Licensed under the Apache License, Version 2.0 (the "License");
     6  // you may not use this file except in compliance with the License.
     7  // You may obtain a copy of the License at
     8  //
     9  //   http://www.apache.org/licenses/LICENSE-2.0
    10  //
    11  // Unless required by applicable law or agreed to in writing, software
    12  // distributed under the License is distributed on an "AS IS" BASIS,
    13  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  // See the License for the specific language governing permissions and
    15  // limitations under the License.
    16  // =============================================================================
    17  //
    18  // This file implements a pass to convert MLIR standard ops into the SPIR-V
    19  // ops. It does not legalize FuncOps.
    20  //
    21  //===----------------------------------------------------------------------===//
    22  
    23  #include "mlir/Conversion/StandardToSPIRV/ConvertStandardToSPIRV.h"
    24  #include "mlir/Dialect/SPIRV/Passes.h"
    25  #include "mlir/Dialect/SPIRV/SPIRVDialect.h"
    26  
    27  using namespace mlir;
    28  
    29  namespace {
    30  /// A pass converting MLIR Standard operations into the SPIR-V dialect.
    31  class ConvertStandardToSPIRVPass
    32      : public ModulePass<ConvertStandardToSPIRVPass> {
    33    void runOnModule() override;
    34  };
    35  } // namespace
    36  
    37  void ConvertStandardToSPIRVPass::runOnModule() {
    38    OwningRewritePatternList patterns;
    39    auto module = getModule();
    40  
    41    populateStandardToSPIRVPatterns(module.getContext(), patterns);
    42    ConversionTarget target(*(module.getContext()));
    43    target.addLegalDialect<spirv::SPIRVDialect>();
    44    target.addLegalOp<FuncOp>();
    45  
    46    if (failed(applyPartialConversion(module, target, patterns))) {
    47      return signalPassFailure();
    48    }
    49  }
    50  
    51  std::unique_ptr<ModulePassBase>
    52  mlir::spirv::createConvertStandardToSPIRVPass() {
    53    return std::make_unique<ConvertStandardToSPIRVPass>();
    54  }
    55  
    56  static PassRegistration<ConvertStandardToSPIRVPass>
    57      pass("convert-std-to-spirv", "Convert Standard Ops to SPIR-V dialect");