| 39 | } |
| 40 | |
| 41 | Maybe<Symbol<SbpParallel>> GetSplitSbpParallel(int axis) { |
| 42 | CHECK_GE_OR_RETURN(axis, 0) << Error::RuntimeError() |
| 43 | << "Split axis must not be negative, but got " << axis << "!"; |
| 44 | CHECK_LT_OR_RETURN(axis, kMaxSplitAxis) |
| 45 | << Error::RuntimeError() << "Expected split axis to be less than the supported maximum axis (" |
| 46 | << kMaxSplitAxis << "), but got " << axis << "!"; |
| 47 | static std::vector<Symbol<SbpParallel>> split_sbp_sym_list = |
| 48 | *JUST(MakeSplitSbpParallelList(kMaxSplitAxis)); |
| 49 | return split_sbp_sym_list.at(axis); |
| 50 | } |
| 51 | |
| 52 | Maybe<Symbol<SbpParallel>> GetBroadcastSbpParallel() { |
| 53 | static Symbol<SbpParallel> broadcast_sbp = JUST(MakeBroadcastSbpParallel()); |
no test coverage detected