contrib/llvm/lib/CodeGen/LowLevelType.cpp

   1 //===-- llvm/CodeGen/GlobalISel/LowLevelType.cpp --------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 /// \file This file implements the more header-heavy bits of the LLT class to
  11 /// avoid polluting users' namespaces.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/CodeGen/LowLevelType.h"
  16 #include "llvm/IR/DataLayout.h"
  17 #include "llvm/IR/DerivedTypes.h"
  18 #include "llvm/Support/raw_ostream.h"
  19 using namespace llvm;
  20
  21 LLT::LLT(Type &Ty, const DataLayout &DL) {
  22   if (auto VTy = dyn_cast<VectorType>(&Ty)) {
  23     SizeInBits = VTy->getElementType()->getPrimitiveSizeInBits();
  24     ElementsOrAddrSpace = VTy->getNumElements();
  25     Kind = ElementsOrAddrSpace == 1 ? Scalar : Vector;
  26   } else if (auto PTy = dyn_cast<PointerType>(&Ty)) {
  27     Kind = Pointer;
  28     SizeInBits = DL.getTypeSizeInBits(&Ty);
  29     ElementsOrAddrSpace = PTy->getAddressSpace();
  30   } else if (Ty.isSized()) {
  31     // Aggregates are no different from real scalars as far as GlobalISel is
  32     // concerned.
  33     Kind = Scalar;
  34     SizeInBits = DL.getTypeSizeInBits(&Ty);
  35     ElementsOrAddrSpace = 1;
  36     assert(SizeInBits != 0 && "invalid zero-sized type");
  37   } else {
  38     Kind = Invalid;
  39     SizeInBits = ElementsOrAddrSpace = 0;
  40   }
  41 }
  42
  43 LLT::LLT(MVT VT) {
  44   if (VT.isVector()) {
  45     SizeInBits = VT.getVectorElementType().getSizeInBits();
  46     ElementsOrAddrSpace = VT.getVectorNumElements();
  47     Kind = ElementsOrAddrSpace == 1 ? Scalar : Vector;
  48   } else if (VT.isValid()) {
  49     // Aggregates are no different from real scalars as far as GlobalISel is
  50     // concerned.
  51     Kind = Scalar;
  52     SizeInBits = VT.getSizeInBits();
  53     ElementsOrAddrSpace = 1;
  54     assert(SizeInBits != 0 && "invalid zero-sized type");
  55   } else {
  56     Kind = Invalid;
  57     SizeInBits = ElementsOrAddrSpace = 0;
  58   }
  59 }
  60
  61 void LLT::print(raw_ostream &OS) const {
  62   if (isVector())
  63     OS << "<" << ElementsOrAddrSpace << " x s" << SizeInBits << ">";
  64   else if (isPointer())
  65     OS << "p" << getAddressSpace();
  66   else if (isValid()) {
  67     assert(isScalar() && "unexpected type");
  68     OS << "s" << getScalarSizeInBits();
  69   } else
  70     llvm_unreachable("trying to print an invalid type");
  71 }