RTGPassPipelines.cpp

Go to the documentation of this file.

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#include "circt/Dialect/RTG/Transforms/RTGPassPipelines.h"
#include "circt/Dialect/Emit/EmitOps.h"
#include "circt/Dialect/RTG/IR/RTGOps.h"
#include "circt/Dialect/RTG/Transforms/RTGPasses.h"
#include "circt/Support/Passes.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Transforms/Passes.h"
 
using namespace mlir;
 
//===----------------------------------------------------------------------===//
// Pipeline implementation.
//===----------------------------------------------------------------------===//
 
void circt::rtg::populateRandomizationPipeline(
    OpPassManager &pm, const RandomizationPipelineOptions &options) {
  {
    ElaborationPassOptions passOptions;
    passOptions.seed = options.seed;
    pm.addPass(rtg::createElaborationPass(passOptions));
  }
  pm.addPass(rtg::createInlineSequencesPass());
  pm.addPass(createSymbolDCEPass());
  {
    auto &testPm = pm.nest<rtg::TestOp>();
    MemoryAllocationPassOptions passOptions;
    passOptions.useImmediates = options.memoriesAsImmediates;
    testPm.addPass(rtg::createMemoryAllocationPass(passOptions));
  }
}
 
void circt::rtg::populateEmissionPipeline(
    OpPassManager &pm, const EmissionPipelineOptions &options) {
  {
    InsertTestToFileMappingPassOptions passOptions;
    passOptions.splitOutput = options.splitOutput;
    passOptions.path = options.path;
    pm.addPass(rtg::createInsertTestToFileMappingPass(passOptions));
  }
  pm.addPass(rtg::createSimpleTestInlinerPass());
  {
    auto &testPm = pm.nest<emit::FileOp>();
    testPm.addPass(rtg::createUniqueValidateOpsPass());
    testPm.addPass(rtg::createLinearScanRegisterAllocationPass());
  }
  {
    auto &anyPm = pm.nestAny();
    anyPm.addPass(mlir::createCSEPass());
    anyPm.addPass(createSimpleCanonicalizerPass());
  }
  pm.addPass(rtg::createLowerValidateToLabelsPass());
  {
    auto &filePm = pm.nest<emit::FileOp>();
    filePm.addPass(rtg::createLowerUniqueLabelsPass());
    filePm.addPass(rtg::createEmitRTGISAAssemblyPass());
  }
}
 
//===----------------------------------------------------------------------===//
// Pipeline registration.
//===----------------------------------------------------------------------===//
 
void circt::rtg::registerPipelines() {
  PassPipelineRegistration<RandomizationPipelineOptions>(
      "rtg-randomization-pipeline",
      "The default pipeline for randomizing RTG tests.",
      populateRandomizationPipeline);
  PassPipelineRegistration<EmissionPipelineOptions>(
      "rtg-emission-pipeline", "The default pipeline for emitting RTG tests.",
      populateEmissionPipeline);
}