Method readBytecode

lib/Bytecode/Reader/BytecodeReader.cpp:2461–2621

}

OwningOpRef<cuda_tile::ModuleOp>
cuda_tile::readBytecode(llvm::MemoryBufferRef bytecodeBuffer,
                        MLIRContext &context) {
  ArrayRef<uint8_t> bytecodeData(
      reinterpret_cast<const uint8_t *>(bytecodeBuffer.getBufferStart()),
      bytecodeBuffer.getBufferSize());

  EncodingReader reader(bytecodeData, context);
  DebugInfoReader debuginfo(context, reader);

  BytecodeVersion bytecodeVersion;
  if (failed(parseHeader(reader, context, bytecodeVersion)))
    return reader.emitError() << "failed to parse bytecode header", nullptr;

  // Store section payloads to allow parsing in a specific order later.
  std::array<std::optional<ArrayRef<uint8_t>>, Section::NumSections + 1>
      sectionPayloads;

  // Discover all sections and store their payloads.
  while (true) {
    SectionHeader header;
    if (failed(parseSectionHeader(reader, header, context)))
      return reader.emitError() << "failed to parse section header", nullptr;

    // Check for the end of the bytecode stream.
    if (header.sectionID == Section::EndOfBytecode) {
      if (reader.remaining() != 0) {
        reader.emitError() << "end section is not the last section";
        return nullptr;
      }
      break;
    }

    auto validateSectionAlignment = [&](uint64_t requiredAlignment) -> bool {
      if (!header.hasAlignment || (header.alignment % requiredAlignment) != 0) {
        reader.emitError() << "section " << static_cast<int>(header.sectionID)
                           << " must have alignment that is a multiple of "
                           << requiredAlignment << " bytes, but "
                           << (header.hasAlignment
                                   ? ("has alignment of " +
                                      std::to_string(header.alignment))
                                   : "has no alignment flag set");
        return false;
      }
      return true;
    };

    switch (header.sectionID) {
    case Section::String:
    case Section::Type:
      if (!validateSectionAlignment(alignof(uint32_t)))
        return nullptr;
      break;
    case Section::Constant:
    case Section::Debug:
    case Section::Func:
      if (!validateSectionAlignment(alignof(uint64_t)))
        return nullptr;