iree/compiler/Dialect/VM/Conversion/VMToEmitC/ConvertVMToEmitC.h

// Copyright 2020 The IREE Authors
//
// Licensed 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

#ifndef IREE_COMPILER_DIALECT_VM_CONVERSION_VMTOEMITC_CONVERTVMTOEMITC_H_
#define IREE_COMPILER_DIALECT_VM_CONVERSION_VMTOEMITC_CONVERTVMTOEMITC_H_

#include "iree/compiler/Dialect/VM/Analysis/RegisterAllocation.h"
#include "iree/compiler/Dialect/VM/Analysis/ValueLiveness.h"
#include "iree/compiler/Dialect/VM/Conversion/VMToEmitC/EmitCTypeConverter.h"
#include "iree/compiler/Dialect/VM/IR/VMOps.h"
#include "mlir/Pass/Pass.h"

namespace mlir {
namespace iree_compiler {

struct VMAnalysis {
 public:
  VMAnalysis(RegisterAllocation &&registerAllocation,
             ValueLiveness &&valueLiveness)
      : registerAllocation(std::move(registerAllocation)),
        valueLiveness(std::move(valueLiveness)) {}

  VMAnalysis(VMAnalysis &&) = default;
  VMAnalysis &operator=(VMAnalysis &&) = default;
  VMAnalysis(const VMAnalysis &) = delete;
  VMAnalysis &operator=(const VMAnalysis &) = delete;

  int getNumRefRegisters() {
    return registerAllocation.getMaxRefRegisterOrdinal() + 1;
  }

  int getRefRegisterOrdinal(Value ref) {
    return registerAllocation.mapToRegister(originalValue(ref)).ordinal();
  }

  bool isLastValueUse(Value ref, Operation *op) {
    return valueLiveness.isLastValueUse(originalValue(ref), op);
  }

  void remapValue(Value original, Value replacement) {
    mapping[replacement] = original;
    return;
  }

 private:
  RegisterAllocation registerAllocation;
  ValueLiveness valueLiveness;
  DenseMap<Value, Value> mapping;

  Value originalValue(Value ref) {
    auto ptr = mapping.find(ref);
    return ptr == mapping.end() ? ref : ptr->second;
  }
};

using VMAnalysisCache = DenseMap<Operation *, VMAnalysis>;

void populateVMToEmitCPatterns(MLIRContext *context,
                               IREE::VM::EmitCTypeConverter &typeConverter,
                               OwningRewritePatternList &patterns,
                               VMAnalysisCache &vmAnalysisCache);

namespace IREE {
namespace VM {

std::unique_ptr<OperationPass<IREE::VM::ModuleOp>> createConvertVMToEmitCPass();

}  // namespace VM
}  // namespace IREE

}  // namespace iree_compiler
}  // namespace mlir

#endif  // IREE_COMPILER_DIALECT_VM_CONVERSION_VMTOEMITC_CONVERTVMTOEMITC_H_