Method extractVersionFromOperand

tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:86–127

}

std::pair<std::string, std::string>
mlir::tblgen::extractVersionFromOperand(unsigned operandIndex,
                                        const Operator &op) {
  assert(operandIndex < static_cast<unsigned>(op.getNumOperands()) &&
         "TableGen backend bug: operand index out of bounds");

  // Find argument index by scanning for operands only.
  uint32_t currentOperandIndex = 0;
  for (int argIndex = 0; argIndex < op.getNumArgs(); ++argIndex) {
    auto argToOpOrAttr = op.getArgToOperandAttrOrProp(argIndex);

    // Check if this argument is an operand.
    if (argToOpOrAttr.kind() == Operator::OperandAttrOrProp::Kind::Operand) {
      if (currentOperandIndex == operandIndex) {
        // Found the argument index for this operand - get its decorators.
        for (const auto &decorator : op.getArgDecorators(argIndex)) {
          if (!decorator.getDef().isSubClassOf("CudaTileArgMetadata"))
            continue;

          const std::string version =
              decorator.getDef().getValueAsString("sinceVersion").str();
          return parseVersion(version);
        }

        // Found operand but no metadata.
        const auto &operand = op.getOperand(operandIndex);
        PrintFatalError(
            op.getLoc(),
            "operand '" + operand.name.str() + "' in operation '" +
                op.getOperationName() +
                "' is missing version metadata (CudaTileArgMetadata)");
      }
      ++currentOperandIndex;
    }
  }

  // Operand not found in operation arguments.
  const auto &operand = op.getOperand(operandIndex);
  PrintFatalError(op.getLoc(), "operand '" + operand.name.str() +
                                   "' not found in operation '" +
                                   op.getOperationName() + "'");
}

std::pair<StringMap<size_t>, std::optional<std::pair<std::string, std::string>>>
mlir::tblgen::getVersionOrderedBitAssignments(const Operator &op) {