mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 07:26:09 +00:00
0cb6d2f643
This is in preparation for using the same convergence verifier for both LLVM IR and Machine IR. Reviewed By: yassingh Differential Revision: https://reviews.llvm.org/D158394
70 lines
2.1 KiB
C++
70 lines
2.1 KiB
C++
//===- ConvergenceVerifier.cpp - Verify convergence control -----*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/IR/ConvergenceVerifier.h"
|
|
#include "llvm/IR/Dominators.h"
|
|
#include "llvm/IR/GenericConvergenceVerifierImpl.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/SSAContext.h"
|
|
|
|
using namespace llvm;
|
|
|
|
template <>
|
|
const Instruction *
|
|
GenericConvergenceVerifier<SSAContext>::findAndCheckConvergenceTokenUsed(
|
|
const Instruction &I) {
|
|
auto *CB = dyn_cast<CallBase>(&I);
|
|
if (!CB)
|
|
return nullptr;
|
|
|
|
unsigned Count =
|
|
CB->countOperandBundlesOfType(LLVMContext::OB_convergencectrl);
|
|
CheckOrNull(Count <= 1,
|
|
"The 'convergencectrl' bundle can occur at most once on a call",
|
|
{Context.print(CB)});
|
|
if (!Count)
|
|
return nullptr;
|
|
|
|
auto Bundle = CB->getOperandBundle(LLVMContext::OB_convergencectrl);
|
|
CheckOrNull(Bundle->Inputs.size() == 1 &&
|
|
Bundle->Inputs[0]->getType()->isTokenTy(),
|
|
"The 'convergencectrl' bundle requires exactly one token use.",
|
|
{Context.print(CB)});
|
|
auto *Token = Bundle->Inputs[0].get();
|
|
auto *Def = dyn_cast<Instruction>(Token);
|
|
|
|
CheckOrNull(
|
|
Def && isConvergenceControlIntrinsic(SSAContext::getIntrinsicID(*Def)),
|
|
"Convergence control tokens can only be produced by calls to the "
|
|
"convergence control intrinsics.",
|
|
{Context.print(Token), Context.print(&I)});
|
|
|
|
if (Def)
|
|
Tokens[&I] = Def;
|
|
|
|
return Def;
|
|
}
|
|
|
|
template <>
|
|
bool GenericConvergenceVerifier<SSAContext>::isInsideConvergentFunction(
|
|
const InstructionT &I) {
|
|
auto *F = I.getFunction();
|
|
return F->isConvergent();
|
|
}
|
|
|
|
template <>
|
|
bool GenericConvergenceVerifier<SSAContext>::isConvergent(
|
|
const InstructionT &I) {
|
|
if (auto *CB = dyn_cast<CallBase>(&I)) {
|
|
return CB->isConvergent();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
template class llvm::GenericConvergenceVerifier<SSAContext>;
|