Selected stage: loom-lower-forall-to-thread. Test exactly output obligation mlir-obligation:d1dc2174ff936b5d9497d7eb with its governing context; report any stag

24                         std::optional<uint64_t> workItemArgOrdinal) {25  switch (kind) {26  case ThreadDomainKind::DenseRectangular:false branch27    if (workItemArgOrdinal)28      return emitError()⋮30                "ordinal";31    return success();32  case ThreadDomainKind::DynamicWork:true branch33    if (!workItemArgOrdinal)false branch34      return emitError()35             << "dynamic-work thread domain requires a work-item argument "36                "ordinal";37    return success();38  }39  return emitError() << "unknown thread domain kind";

102enum class ExtentExprKind { Unsupported, Constant, AddI, IndexCast };103104static bool isScalarNonzeroSignlessIntegerOrIndex(Type type) {105  if (isa<IndexType>(type))true branch106    return true;107  auto integerType = dyn_cast<IntegerType>(type);108  return integerType && integerType.isSignless() && integerType.getWidth() != 0;109}110111static unsigned getScalarIntegerBitWidth(Type type) {⋮115}116117static ExtentExprKind classifyExtentExpr(Operation *op) {118  if (op->getNumRegions() != 0 || op->getNumSuccessors() != 0)false branchfalse branch119    return ExtentExprKind::Unsupported;120121  if (isa<arith::ConstantOp>(op)) {true branch122    if (op->getNumOperands() != 0 || op->getNumResults() != 1)false branchfalse branch123      return ExtentExprKind::Unsupported;124    Type resultType = op->getResult(0).getType();125    auto value = dyn_cast_or_null<IntegerAttr>(126        cast<arith::ConstantOp>(op).getProperties().value);127    if (!isScalarNonzeroSignlessIntegerOrIndex(resultType) || !value ||false branchfalse branchfalse branch128        value.getType() != resultType)false branch129      return ExtentExprKind::Unsupported;130    return ExtentExprKind::Constant;131  }132133  if (isa<arith::AddIOp>(op)) {⋮170    SmallVector<Frame> stack;171    schedule(value, stack);172    while (!stack.empty()) {true branch173      Frame &frame = stack.back();174      if (constants.contains(frame.value)) {false branch175        active.erase(frame.value);176        stack.pop_back();⋮178      }179180      Operation *definingOp = cast<OpResult>(frame.value).getOwner();181      bool descended = false;182      while (frame.nextOperand < definingOp->getNumOperands()) {false branch183        Value operand = definingOp->getOperand(frame.nextOperand++);184        if (constants.contains(operand))⋮189        }190      }191      if (descended)false branch192        continue;193194      evaluateOperation(definingOp, frame.kind);195      active.erase(frame.value);196      stack.pop_back();197    }198199    return constants.lookup(value);⋮217218    auto result = dyn_cast<OpResult>(value);219    if (!result) {false branch220      constants.try_emplace(value, Attribute{});221      return false;222    }223224    Operation *definingOp = result.getOwner();225    ExtentExprKind kind = classifyExtentExpr(definingOp);226    if (kind == ExtentExprKind::Unsupported) {false branch227      cacheUnknown(definingOp);228      return false;229    }230    if (!active.insert(value).second) {false branch231      cacheUnknown(definingOp);232      return false;233    }234235    stack.push_back({value, 0, kind});236    return true;237  }238239  void evaluateOperation(Operation *op, ExtentExprKind kind) {240    if (kind == ExtentExprKind::Constant) {true branch241      constants.try_emplace(242          op->getResult(0),243          cast<IntegerAttr>(cast<arith::ConstantOp>(op).getProperties().value));244      return;245    }246247    auto operandConstant = [&](unsigned index) {⋮329          auto constant =330              dyn_cast_or_null<IntegerAttr>(evaluator.evaluate(extent));331          if (constant && constant.getValue().isNegative()) {true branchfalse branch332            launch.emitOpError("grid upper bound #")333                << index << " must be nonnegative";⋮471    if (parseOne())472      return failure();473    while (succeeded(parser.parseOptionalComma()))true branch474      if (parseOne())false branch475        return failure();476    return parser.parseRParen();⋮539      p << " iv (";540      for (size_t i = N + 1, e = entry->getNumArguments(); i < e; ++i) {541        if (i > N + 1)true branch542          p << ", ";543        p.printRegionArgument(entry->getArgument(i));544      }⋮567    return emitOpError("must not declare function results");568569  if (getDomain().getKind() == ThreadDomainKind::DynamicWork) {true branch570    uint64_t ordinal = *getDomain().getWorkItemArgOrdinal();571    ArrayRef<Type> inputs = getFunctionType().getInputs();572    if (ordinal >= inputs.size())false branch573      return emitOpError("dynamic-work item argument ordinal ")574             << ordinal << " is out of bounds for " << inputs.size()575             << " thread inputs";576    for (auto [index, type] : llvm::enumerate(inputs))false branchtrue branch577      if (DataflowDialect::containsChannelOrThreadToken(type))false branch578        return emitOpError("dynamic-work thread input #")579               << index << " must not contain a channel or thread token";580  }581  if (!ownsThreadLaunchExtentAnalysis(*this))582    return success();⋮654//===----------------------------------------------------------------------===//655656LogicalResult ThreadWaitOp::verify() {657  if ((*this)->getParentOfType<ThreadOp>() ||false branchfalse branch658      (*this)->getParentOfType<GraphOp>())false branch659    return emitOpError(660        "must appear outside any dataflow.thread or dataflow.graph "661        "definition");662663  for (auto [index, token] : llvm::enumerate(getAsyncDependencies()))false branchtrue branch664    if (!token.getDefiningOp<ThreadLaunchOp>())false branch665      return emitOpError("operand #") << index666                                      << " must be produced directly by "667                                         "dataflow.thread.launch";668  return success();669}670671//===----------------------------------------------------------------------===//

45      rejected = true;46    });47    if (rejected)false branch48      signalPassFailure();49  }