89 lines
2.8 KiB
C++
89 lines
2.8 KiB
C++
//===- AMDGPUResourceUsageAnalysis.h ---- analysis of resources -*- 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
|
|
/// \brief Analyzes how many registers and other resources are used by
|
|
/// functions.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPURESOURCEUSAGEANALYSIS_H
|
|
#define LLVM_LIB_TARGET_AMDGPU_AMDGPURESOURCEUSAGEANALYSIS_H
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/CodeGen/MachineFunctionPass.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
|
|
namespace llvm {
|
|
|
|
class GCNSubtarget;
|
|
class MachineFunction;
|
|
class GCNTargetMachine;
|
|
|
|
struct AMDGPUResourceUsageAnalysisImpl {
|
|
public:
|
|
static char ID;
|
|
// Track resource usage for callee functions.
|
|
struct SIFunctionResourceInfo {
|
|
// Track the number of explicitly used VGPRs. Special registers reserved at
|
|
// the end are tracked separately.
|
|
int32_t NumVGPR = 0;
|
|
int32_t NumAGPR = 0;
|
|
int32_t NumExplicitSGPR = 0;
|
|
uint64_t CalleeSegmentSize = 0;
|
|
uint64_t PrivateSegmentSize = 0;
|
|
bool UsesVCC = false;
|
|
bool UsesFlatScratch = false;
|
|
bool HasDynamicallySizedStack = false;
|
|
bool HasRecursion = false;
|
|
bool HasIndirectCall = false;
|
|
SmallVector<const Function *, 16> Callees;
|
|
};
|
|
|
|
SIFunctionResourceInfo
|
|
analyzeResourceUsage(const MachineFunction &MF,
|
|
uint32_t AssumedStackSizeForDynamicSizeObjects,
|
|
uint32_t AssumedStackSizeForExternalCall) const;
|
|
};
|
|
|
|
struct AMDGPUResourceUsageAnalysisWrapperPass : public MachineFunctionPass {
|
|
using FunctionResourceInfo =
|
|
AMDGPUResourceUsageAnalysisImpl::SIFunctionResourceInfo;
|
|
FunctionResourceInfo ResourceInfo;
|
|
|
|
public:
|
|
static char ID;
|
|
AMDGPUResourceUsageAnalysisWrapperPass() : MachineFunctionPass(ID) {}
|
|
|
|
bool runOnMachineFunction(MachineFunction &MF) override;
|
|
|
|
const FunctionResourceInfo &getResourceInfo() const { return ResourceInfo; }
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.setPreservesAll();
|
|
MachineFunctionPass::getAnalysisUsage(AU);
|
|
}
|
|
};
|
|
|
|
class AMDGPUResourceUsageAnalysis
|
|
: public AnalysisInfoMixin<AMDGPUResourceUsageAnalysis> {
|
|
friend AnalysisInfoMixin<AMDGPUResourceUsageAnalysis>;
|
|
static AnalysisKey Key;
|
|
|
|
const GCNTargetMachine &TM;
|
|
|
|
public:
|
|
using Result = AMDGPUResourceUsageAnalysisImpl::SIFunctionResourceInfo;
|
|
Result run(MachineFunction &MF, MachineFunctionAnalysisManager &MFAM);
|
|
|
|
AMDGPUResourceUsageAnalysis(const GCNTargetMachine &TM_) : TM(TM_) {}
|
|
};
|
|
|
|
} // namespace llvm
|
|
#endif // LLVM_LIB_TARGET_AMDGPU_AMDGPURESOURCEUSAGEANALYSIS_H
|