TargetInfo.cpp

Go to the documentation of this file.

#include "TargetInfo.h"
#include "ABIInfo.h"
#include "CIRGenFunction.h"
#include "CIRGenModule.h"
#include "mlir/Dialect/Ptr/IR/MemorySpaceInterfaces.h"
#include "clang/Basic/AddressSpaces.h"
#include "clang/CIR/Dialect/IR/CIRAttrs.h"
#include "clang/CIR/Dialect/IR/CIRDialect.h"
#include "clang/CIR/MissingFeatures.h"
#include "clang/CodeGenUtils/TargetUtils.h"
 
using namespace clang;
using namespace clang::CIRGen;
 
bool clang::CIRGen::isEmptyRecordForABI(const ASTContext &context, QualType t) {
  const auto *rd = t->getAsRecordDecl();
  if (!rd)
    return false;
  if (rd->hasFlexibleArrayMember())
    return false;
 
  if (const auto *cxxrd = dyn_cast<CXXRecordDecl>(rd)) {
    // A vtable pointer is neither a base nor a field, so clang's predicate
    // calls a polymorphic class empty and leans on its callers rejecting one as
    // non-trivially-copyable beforehand.  This answer is read off the record
    // type without that precondition, so rule it out here instead.
    if (cxxrd->isDynamicClass())
      return false;
 
    for (const auto &i : cxxrd->bases())
      if (!isEmptyRecordForABI(context, i.getType()))
        return false;
  }
 
  for (const auto *i : rd->fields())
    if (!isEmptyFieldForABI(context, i))
      return false;
  return true;
}
 
bool clang::CIRGen::isEmptyFieldForABI(const ASTContext &context,
                                       const FieldDecl *fd) {
  // A zero-width bit-field holds no bits, so it holds no data.  Any other
  // unnamed bit-field is real storage that the x86-64 classifier gives the
  // eightbyte classes of a named bit-field, so it does hold data.
  //
  // FIXME: this needs an ABI compatibility check.  Clang used to ignore every
  // unnamed bit-field here, and the older compatibility levels have to keep
  // that answer once there is a switch to ask.
  if (fd->isUnnamedBitField())
    return fd->isZeroLengthBitField();
 
  QualType ft = fd->getType();
 
  // An array of empty records is empty, and a zero-length array always is.
  bool wasArray = false;
  while (const ConstantArrayType *at = context.getAsConstantArrayType(ft)) {
    if (at->isZeroSize())
      return true;
    ft = at->getElementType();
    wasArray = true;
  }
 
  const auto *rt = ft->getAsCanonical<RecordType>();
  if (!rt)
    return false;
 
  // A C++ record field is never empty under the Itanium ABI unless
  // [[no_unique_address]] makes it so, and that exception covers a record
  // rather than an array of them.
  if (isa<CXXRecordDecl>(rt->getDecl()) &&
      (wasArray || !fd->hasAttr<NoUniqueAddressAttr>()))
    return false;
 
  return isEmptyRecordForABI(context, ft);
}
 
namespace {
 
class AMDGPUABIInfo : public ABIInfo {
public:
  AMDGPUABIInfo(CIRGenTypes &cgt) : ABIInfo(cgt) {}
};
 
class AMDGPUTargetCIRGenInfo : public TargetCIRGenInfo {
public:
  AMDGPUTargetCIRGenInfo(CIRGenTypes &cgt)
      : TargetCIRGenInfo(std::make_unique<AMDGPUABIInfo>(cgt)) {}
 
  bool supportsLibCall() const override { return false; }
 
  cir::CallingConv getDeviceKernelCallingConv() const override {
    return cir::CallingConv::AMDGPUKernel;
  }
 
  void setCUDAKernelCallingConvention(const FunctionType *&ft) const override {
    ft = getABIInfo().cgt.getASTContext().adjustFunctionType(
        ft, ft->getExtInfo().withCallingConv(CC_DeviceKernel));
  }
 
  void setTargetAttributes(const clang::Decl *decl, mlir::Operation *global,
                           CIRGenModule &cgm) const override {
    if (auto func = mlir::dyn_cast<cir::FuncOp>(global)) {
      if (CodeGenUtils::requiresAMDGPUProtectedVisibility(
              decl,
              func.getGlobalVisibility() == cir::VisibilityKind::Hidden)) {
        func.setGlobalVisibility(cir::VisibilityKind::Protected);
        func.setDSOLocal(true);
      }
      setAMDGPUTargetFunctionAttributes(decl, func, cgm);
    } else if (auto gv = mlir::dyn_cast<cir::GlobalOp>(global)) {
      if (CodeGenUtils::requiresAMDGPUProtectedVisibility(
              decl, gv.getGlobalVisibility() == cir::VisibilityKind::Hidden)) {
        gv.setGlobalVisibility(cir::VisibilityKind::Protected);
        gv.setDSOLocal(true);
      }
    }
  }
 
  clang::LangAS
  getGlobalVarAddressSpace(CIRGenModule &cgm,
                           const clang::VarDecl *decl) const override {
    using clang::LangAS;
    assert(!cgm.getLangOpts().OpenCL &&
           !(cgm.getLangOpts().CUDA && cgm.getLangOpts().CUDAIsDevice) &&
           "Address space agnostic languages only");
    LangAS defaultGlobalAS = LangAS::opencl_global;
    if (!decl)
      return defaultGlobalAS;
 
    LangAS addrSpace = decl->getType().getAddressSpace();
    if (addrSpace != LangAS::Default)
      return addrSpace;
 
    // Only promote to address space 4 if VarDecl has constant initialization.
    if (decl->getType().isConstantStorage(cgm.getASTContext(), false, false) &&
        decl->hasConstantInitialization())
      return LangAS::opencl_constant;
 
    return defaultGlobalAS;
  }
 
  mlir::ptr::MemorySpaceAttrInterface
  getCIRAllocaAddressSpace() const override {
    return cir::LangAddressSpaceAttr::get(
        &getABIInfo().cgt.getMLIRContext(),
        cir::LangAddressSpace::OffloadPrivate);
  }
};
 
} // namespace
 
namespace {
 
class X8664ABIInfo : public ABIInfo {
public:
  X8664ABIInfo(CIRGenTypes &cgt) : ABIInfo(cgt) {}
};
 
class X8664TargetCIRGenInfo : public TargetCIRGenInfo {
public:
  X8664TargetCIRGenInfo(CIRGenTypes &cgt)
      : TargetCIRGenInfo(std::make_unique<X8664ABIInfo>(cgt)) {}
};
} // namespace
 
std::unique_ptr<TargetCIRGenInfo>
clang::CIRGen::createAMDGPUTargetCIRGenInfo(CIRGenTypes &cgt) {
  return std::make_unique<AMDGPUTargetCIRGenInfo>(cgt);
}
 
std::unique_ptr<TargetCIRGenInfo>
clang::CIRGen::createX8664TargetCIRGenInfo(CIRGenTypes &cgt) {
  return std::make_unique<X8664TargetCIRGenInfo>(cgt);
}
 
ABIInfo::~ABIInfo() noexcept = default;
 
bool TargetCIRGenInfo::isNoProtoCallVariadic(
    const FunctionNoProtoType *fnType) const {
  // The following conventions are known to require this to be false:
  //   x86_stdcall
  //   MIPS
  // For everything else, we just prefer false unless we opt out.
  return false;
}
 
cir::CallingConv TargetCIRGenInfo::getDeviceKernelCallingConv() const {
  // Device kernels are entered through a runtime API, not called as normal
  // sub-functions, so a modified C calling convention is used.
  assert(getABIInfo().cgt.getASTContext().getLangOpts().OpenCL &&
         "Kernel calling convention only defined for OpenCL");
  return cir::CallingConv::C;
}
 
mlir::Value TargetCIRGenInfo::getNullPointer(CIRGenModule &cgm,
                                             cir::PointerType ptrTy,
                                             QualType qt,
                                             mlir::Location loc) const {
  return cgm.getBuilder().getNullPtr(ptrTy, loc);
}
 
clang::LangAS
TargetCIRGenInfo::getGlobalVarAddressSpace(CIRGenModule &cgm,
                                           const clang::VarDecl *d) const {
  assert(!cgm.getLangOpts().OpenCL &&
         !(cgm.getLangOpts().CUDA && cgm.getLangOpts().CUDAIsDevice) &&
         "Address space agnostic languages only");
  return d ? d->getType().getAddressSpace() : LangAS::Default;
}