llvm.org GIT mirror llvm / master lib / Target / AMDGPU / AMDGPUMachineFunction.h

Tree @master (Download .tar.gz)

AMDGPUMachineFunction.h @masterraw · history · blame

//===-- AMDGPUMachineFunctionInfo.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


#include "llvm/ADT/DenseMap.h"
#include "llvm/CodeGen/MachineFunction.h"

namespace llvm {

class GCNSubtarget;

class AMDGPUMachineFunction : public MachineFunctionInfo {
  /// A map to keep track of local memory objects and their offsets within the
  /// local memory space.
  SmallDenseMap<const GlobalValue *, unsigned, 4> LocalMemoryObjects;

  uint64_t ExplicitKernArgSize; // Cache for this.
  Align MaxKernArgAlign;        // Cache for this.

  /// Number of bytes in the LDS that are being used.
  unsigned LDSSize;

  // Kernels + shaders. i.e. functions called by the driver and not called
  // by other functions.
  bool IsEntryFunction;

  bool NoSignedZerosFPMath;

  // Function may be memory bound.
  bool MemoryBound;

  // Kernel may need limited waves per EU for better performance.
  bool WaveLimiter;

  AMDGPUMachineFunction(const MachineFunction &MF);

  uint64_t getExplicitKernArgSize() const {
    return ExplicitKernArgSize;

  unsigned getMaxKernArgAlign() const { return MaxKernArgAlign.value(); }

  unsigned getLDSSize() const {
    return LDSSize;

  bool isEntryFunction() const {
    return IsEntryFunction;

  bool hasNoSignedZerosFPMath() const {
    return NoSignedZerosFPMath;

  bool isMemoryBound() const {
    return MemoryBound;

  bool needsWaveLimiter() const {
    return WaveLimiter;

  unsigned allocateLDSGlobal(const DataLayout &DL, const GlobalValue &GV);