diff --git a/peregrine/test/cluster/node/BUILD b/peregrine/test/cluster/node/BUILD index 501ee78..6d6e694 100644 --- a/peregrine/test/cluster/node/BUILD +++ b/peregrine/test/cluster/node/BUILD @@ -211,5 +211,6 @@ cc_binary( "@abseil-cpp//absl/log:initialize", "@abseil-cpp//absl/log", "@abseil-cpp//absl/log:check", + "@abseil-cpp//absl/log:flags", ], ) diff --git a/peregrine/test/cluster/orchestrator/bipartite_runner.py b/peregrine/test/cluster/orchestrator/bipartite_runner.py index 8abec32..e0d2df6 100644 --- a/peregrine/test/cluster/orchestrator/bipartite_runner.py +++ b/peregrine/test/cluster/orchestrator/bipartite_runner.py @@ -87,6 +87,12 @@ True, "Share one registered buffer across all instances in a node process.", ) +_NODE_MINLOGLEVEL = flags.DEFINE_integer( + "node_minloglevel", + 1, + "Minimum Abseil log level for :node processes (0=INFO, 1=WARNING," + " 2=ERROR, 3=FATAL).", +) _METRICS_INTERVAL = flags.DEFINE_string( "metrics_interval", "1s", @@ -150,6 +156,7 @@ def main(argv: Sequence[str]) -> None: num_xfers=_NUM_XFERS.value, traffic_pattern=_TRAFFIC_PATTERN.value, share_buffer=_SHARE_BUFFER.value, + node_minloglevel=_NODE_MINLOGLEVEL.value, metrics_interval=_METRICS_INTERVAL.value, ssh_cmd=_SSH_CMD.value, remote_dir=_REMOTE_DIR.value, diff --git a/peregrine/test/cluster/orchestrator/cluster_session.py b/peregrine/test/cluster/orchestrator/cluster_session.py index 7524f16..86da3a8 100644 --- a/peregrine/test/cluster/orchestrator/cluster_session.py +++ b/peregrine/test/cluster/orchestrator/cluster_session.py @@ -36,6 +36,7 @@ class BipartiteSessionConfig: num_xfers: int = 1 traffic_pattern: str = "all_to_all" share_buffer: bool = True + node_minloglevel: int = 1 metrics_interval: str = "1s" ssh_cmd: str = node_executor.DEFAULT_SSH_CMD remote_dir: str = "/tmp" @@ -168,6 +169,7 @@ def _common_workload_flags(cfg: BipartiteSessionConfig) -> list[str]: f"--num_xfers={cfg.num_xfers}", f"--traffic_pattern={cfg.traffic_pattern}", f"--share_buffer={'true' if cfg.share_buffer else 'false'}", + f"--minloglevel={cfg.node_minloglevel}", f"--metrics_interval={cfg.metrics_interval}", ] if cfg.workload.lower() == "serial_fixed_write":