cross-project-tests/debuginfo-tests/llvm-prettyprinters/gdb/mlir-support.cpp

   1 #include "mlir/IR/BuiltinAttributes.h"
   2 #include "mlir/IR/BuiltinTypes.h"
   3 #include "mlir/IR/Location.h"
   4 #include "mlir/IR/MLIRContext.h"
   5 #include "mlir/IR/OperationSupport.h"
   6
   7 mlir::MLIRContext Context;
   8
   9 auto Identifier = mlir::StringAttr::get(&Context, "foo");
  10 mlir::OperationName OperationName("FooOp", &Context);
  11
  12 mlir::Type Type(nullptr);
  13 mlir::Type IndexType = mlir::IndexType::get(&Context);
  14 mlir::Type IntegerType =
  15     mlir::IntegerType::get(&Context, 3, mlir::IntegerType::Unsigned);
  16 mlir::Type FloatType = mlir::Float32Type::get(&Context);
  17 mlir::Type MemRefType = mlir::MemRefType::get({4, 5}, FloatType);
  18 mlir::Type UnrankedMemRefType = mlir::UnrankedMemRefType::get(IntegerType, 6);
  19 mlir::Type VectorType = mlir::VectorType::get({1, 2}, FloatType);
  20 mlir::Type TupleType =
  21     mlir::TupleType::get(&Context, mlir::TypeRange({IndexType, FloatType}));
  22
  23
  24 mlir::detail::OutOfLineOpResult Result(FloatType, 42);
  25 mlir::Value Value(&Result);
  26
  27 auto UnknownLoc = mlir::UnknownLoc::get(&Context);
  28 auto FileLineColLoc = mlir::FileLineColLoc::get(&Context, "file", 7, 8);
  29 auto OpaqueLoc = mlir::OpaqueLoc::get<uintptr_t>(9, &Context);
  30 auto NameLoc = mlir::NameLoc::get(Identifier);
  31 auto CallSiteLoc = mlir::CallSiteLoc::get(FileLineColLoc, OpaqueLoc);
  32 auto FusedLoc = mlir::FusedLoc::get(&Context, {FileLineColLoc, NameLoc});
  33
  34 mlir::Attribute UnitAttr = mlir::UnitAttr::get(&Context);
  35 mlir::Attribute FloatAttr = mlir::FloatAttr::get(FloatType, 1.0);
  36 mlir::Attribute IntegerAttr = mlir::IntegerAttr::get(IntegerType, 10);
  37 mlir::Attribute TypeAttr = mlir::TypeAttr::get(IndexType);
  38 mlir::Attribute ArrayAttr = mlir::ArrayAttr::get(&Context, {UnitAttr});
  39 mlir::Attribute StringAttr = mlir::StringAttr::get(&Context, "foo");
  40 mlir::Attribute ElementsAttr = mlir::DenseElementsAttr::get(
  41     VectorType.cast<mlir::ShapedType>(), llvm::ArrayRef<float>{2.0f, 3.0f});
  42
  43 int main() {
  44   // Reference symbols that might otherwise be stripped.
  45   std::uintptr_t result = 0;
  46   auto dont_strip = [&](const auto &val) {
  47     result += reinterpret_cast<std::uintptr_t>(&val);
  48   };
  49   dont_strip(Value);
  50   return result; // Non-zero return value is OK.
  51 }