Function GetSbpFromState

oneflow/api/python/symbol/sbp_symbol.cpp:74–85

}

Maybe<Symbol<SbpParallel>> GetSbpFromState(const std::pair<std::string, int>& state) {
  if (state.first == "B") {
    return GetBroadcastSbpParallel();
  } else if (state.first == "P") {
    return GetPartialSumSbpParallel();
  } else if (state.first == "S") {
    return GetSplitSbpParallel(state.second);
  } else {
    return Error::RuntimeError() << "Invalid sbp signature state: (" << state.first << ", "
                                 << state.second << ");";
  }
}

}  // namespace