6024e5c395
clang 20.0.0 (based on r547379) from build 12806354. Bug: http://b/379133546 Test: N/A Change-Id: I2eb8938af55d809de674be63cb30cf27e801862b Upstream-Commit: ad834e67b1105d15ef907f6255d4c96e8e733f57
85 lines
2.8 KiB
C++
85 lines
2.8 KiB
C++
//===- 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
|