//===- GenericConvergenceVerifier.h ---------------------------*- C++ -*---===//
//
// 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
//
//===----------------------------------------------------------------------===//
///
/// \file
///
/// A verifier for the static rules of convergence control tokens that works
/// with both LLVM IR and MIR.
///
//===----------------------------------------------------------------------===//

#ifndef LLVM_ADT_GENERICCONVERGENCEVERIFIER_H
#define LLVM_ADT_GENERICCONVERGENCEVERIFIER_H

#include "llvm/ADT/GenericCycleInfo.h"

namespace llvm {

template <typename ContextT> class GenericConvergenceVerifier {
public:
  using BlockT = typename ContextT::BlockT;
  using FunctionT = typename ContextT::FunctionT;
  using ValueRefT = typename ContextT::ValueRefT;
  using InstructionT = typename ContextT::InstructionT;
  using DominatorTreeT = typename ContextT::DominatorTreeT;
  using CycleInfoT = GenericCycleInfo<ContextT>;
  using CycleT = typename CycleInfoT::CycleT;

  void initialize(raw_ostream *OS,
                  function_ref<void(const Twine &Message)> FailureCB,
                  const FunctionT &F) {
    clear();
    this->OS = OS;
    this->FailureCB = FailureCB;
    Context = ContextT(&F);
  }

  void clear();
  void visit(const BlockT &BB);
  void visit(const InstructionT &I);
  void verify(const DominatorTreeT &DT);

  bool sawTokens() const { return ConvergenceKind == ControlledConvergence; }

private:
  raw_ostream *OS;
  std::function<void(const Twine &Message)> FailureCB;
  DominatorTreeT *DT;
  CycleInfoT CI;
  ContextT Context;

  /// Whether the current function has convergencectrl operand bundles.
  enum {
    ControlledConvergence,
    UncontrolledConvergence,
    NoConvergence
  } ConvergenceKind = NoConvergence;

  /// The control token operation performed by a convergence control Intrinsic
  /// in LLVM IR, or by a CONVERGENCECTRL* instruction in MIR
  enum ConvOpKind { CONV_ANCHOR, CONV_ENTRY, CONV_LOOP, CONV_NONE };

  // Cache token uses found so far. Note that we track the unique definitions
  // and not the token values.
  DenseMap<const InstructionT *, const InstructionT *> Tokens;

  bool SeenFirstConvOp = false;

  static bool isInsideConvergentFunction(const InstructionT &I);
  static bool isConvergent(const InstructionT &I);
  static ConvOpKind getConvOp(const InstructionT &I);
  void checkConvergenceTokenProduced(const InstructionT &I);
  const InstructionT *findAndCheckConvergenceTokenUsed(const InstructionT &I);

  void reportFailure(const Twine &Message, ArrayRef<Printable> Values);
};

} // end namespace llvm

#endif // LLVM_ADT_GENERICCONVERGENCEVERIFIER_H