diff --git a/bench/exo_bench.py b/bench/exo_bench.py index 83390b0b..15cc3a58 100644 --- a/bench/exo_bench.py +++ b/bench/exo_bench.py @@ -431,7 +431,12 @@ def main() -> int: ap.add_argument( "--skip-pipeline-jaccl", action="store_true", - help="Pipeline jaccl is often pointless, skip by default", + help="Skip pipeline+jaccl placements, as it's often pointless.", + ) + ap.add_argument( + "--skip-tensor-ring", + action="store_true", + help="Skip tensor+ring placements, as it's so slow.", ) ap.add_argument( "--repeat", type=int, default=1, help="Repetitions per (pp,tg) pair." @@ -533,6 +538,16 @@ def main() -> int: ): continue + if ( + args.skip_tensor_ring + and ( + args.instance_meta == "both" + and "ring" in p.get("instance_meta", "").lower() + ) + and (args.sharding == "both" and "tensor" in p.get("sharding", "").lower()) + ): + continue + if args.min_nodes <= n <= args.max_nodes: selected.append(p)