blob: d1c9229bc336220b9bc263b858f2b5ee0d79e9ee [file] [log] [blame]
//===- AMDGPULDSUtils.h - LDS related helper functions -*- C++ -*----------===//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// AMDGPU LDS related helper utility functions.
#include "llvm/ADT/DenseMap.h"
#include "llvm/IR/Constants.h"
namespace llvm {
class ConstantExpr;
namespace AMDGPU {
/// Collect reachable callees for each kernel defined in the module \p M and
/// return collected callees at \p KernelToCallees.
void collectReachableCallees(
Module &M,
DenseMap<Function *, SmallPtrSet<Function *, 8>> &KernelToCallees);
/// For the given LDS global \p GV, visit all its users and collect all
/// non-kernel functions within which \p GV is used and return collected list of
/// such non-kernel functions.
SmallPtrSet<Function *, 8> collectNonKernelAccessorsOfLDS(GlobalVariable *GV);
/// Collect all the instructions where user \p U belongs to. \p U could be
/// instruction itself or it could be a constant expression which is used within
/// an instruction. If \p CollectKernelInsts is true, collect instructions only
/// from kernels, otherwise collect instructions only from non-kernel functions.
DenseMap<Function *, SmallPtrSet<Instruction *, 8>>
getFunctionToInstsMap(User *U, bool CollectKernelInsts);
bool isKernelCC(const Function *Func);
Align getAlign(DataLayout const &DL, const GlobalVariable *GV);
/// \returns true if a given global variable \p GV (or its global users) appear
/// as an use within some instruction (either from kernel or from non-kernel).
bool hasUserInstruction(const GlobalValue *GV);
/// \returns true if an LDS global requires lowering to a module LDS structure
/// if \p F is not given. If \p F is given it must be a kernel and function
/// \returns true if an LDS global is directly used from that kernel and it
/// is safe to replace its uses with a kernel LDS structure member.
bool shouldLowerLDSToStruct(const GlobalVariable &GV,
const Function *F = nullptr);
std::vector<GlobalVariable *> findVariablesToLower(Module &M,
const Function *F = nullptr);
SmallPtrSet<GlobalValue *, 32> getUsedList(Module &M);
/// Replace all uses of constant \p C with instructions in \p F.
void replaceConstantUsesInFunction(ConstantExpr *C, const Function *F);
} // end namespace AMDGPU
} // end namespace llvm