Function generateOpcodeMap

tools/cuda-tile-tblgen/BytecodeGen.cpp:198–229

Generates the opcode map implementation from TableGen records

/// Generates the opcode map implementation from TableGen records
static void generateOpcodeMap(const RecordKeeper &records, raw_ostream &os) {
  emitSourceFileHeader("Generated Opcode Map", os);

  // Get all BytecodeOpcode records.
  auto opcodeRecords = records.getAllDerivedDefinitions("BytecodeOpcode");

  os << "namespace mlir {\n"
     << "namespace cuda_tile {\n"
     << "namespace Bytecode {\n\n"
     << "const llvm::StringMap<Opcode> &getOpcodeMap() {\n"
     << "  static const llvm::StringMap<Opcode> opcodeMap = {\n"
     << "    // === PUBLIC OPERATIONS ===\n"
     << "    // These mappings are FROZEN and must never change for backward\n"
     << "    // compatibility.\n";

  // Generate public operation mappings.
  for (const Record *record : opcodeRecords) {
    if (record->isSubClassOf("PublicOpcode")) {
      const Record *opRecord = record->getValueAsDef("operation");
      Operator op(opRecord);
      os << "    {\"" << op.getOperationName()
         << "\", Opcode::" << op.getCppClassName() << "},\n";
    }
  }

  os << "  };\n"
     << "  return opcodeMap;\n"
     << "}\n\n"
     << "} // namespace Bytecode\n"
     << "} // namespace cuda_tile\n"
     << "} // namespace mlir\n";
}

/// Generates the C++ function signature for the 'write<OpName>' function,
/// which handles serialization for a specific cuda_tile operation.