Method ToString

tensorflow/compiler/xla/service/gpu/thunk_schedule.cc:149–185

}

string ThunkSchedule::ToString() const {
  if (thunk_total_order_.empty()) {
    return "No thunks.";
  }

  const Thunk* thunk_with_longest_kind = *absl::c_max_element(
      thunk_total_order_, [](const Thunk* a, const Thunk* b) {
        return ThunkKindToString(a->kind()).length() <
               ThunkKindToString(b->kind()).length();
      });
  int64 max_thunk_kind_len =
      ThunkKindToString(thunk_with_longest_kind->kind()).length();

  string result = "Total order:\n";
  for (Thunk* thunk : thunk_total_order_) {
    // Write out the thunk kind, padded out to max_thunk_kind_len.
    absl::string_view kind_str = ThunkKindToString(thunk->kind());
    absl::StrAppend(&result, kind_str,
                    string(max_thunk_kind_len - kind_str.length(), ' '), "\t");
    if (thunk->hlo_instruction() != nullptr) {
      absl::StrAppend(&result, thunk->hlo_instruction()->ToString());
    } else {
      absl::StrAppend(&result, "(no HloInstruction)");
    }
    absl::StrAppend(&result, "\n");
  }
  absl::StrAppend(&result, "\nDependencies:\n");
  for (const auto& entry : depends_on_) {
    const Thunk* dependent = entry.first;
    for (const Thunk* dependency : entry.second) {
      absl::StrAppend(&result, "\t", dependent->hlo_instruction()->name(),
                      " depends on ", dependency->hlo_instruction()->name(),
                      "\n");
    }
  }
  return result;
}

}  // namespace gpu
}  // namespace xla