minor fix

2026-08-19 16:33:04 +02:00
parent db8d1c1707
commit add20e56eb
2 changed files with 9 additions and 1 deletions
@@ -841,7 +841,7 @@ static FailureOr<PipelineStageAssignment> assignPipelineStages(
    std::string &error) {
  if (graph.nodes.empty())
    return PipelineStageAssignment {
-      {}, std::vector<size_t>(layout.getStageCount(), 1)};
+      {}, std::vector<size_t>(layout.getStageSizes())};
  std::vector<size_t> tasksByOrder(graph.nodes.size());
  std::iota(tasksByOrder.begin(), tasksByOrder.end(), 0);
  llvm::sort(tasksByOrder, [&](size_t lhs, size_t rhs) {