Function MakeSplitSbpParallelList

oneflow/api/python/symbol/sbp_symbol.cpp:34–39

namespace {

Maybe<std::vector<Symbol<SbpParallel>>> MakeSplitSbpParallelList(int max_split_axis) {
  std::shared_ptr<std::vector<Symbol<SbpParallel>>> ret =
      std::make_shared<std::vector<Symbol<SbpParallel>>>(max_split_axis);
  for (int i = 0; i < max_split_axis; ++i) { ret->at(i) = JUST(MakeSplitSbpParallel(i)); }
  return ret;
}

Maybe<Symbol<SbpParallel>> GetSplitSbpParallel(int axis) {
  CHECK_GE_OR_RETURN(axis, 0) << Error::RuntimeError()