Skip to content

Commit 704eac1

Browse files
author
pv-nvidia
committed
Add --ovrtx_async_rendering to the RSL-RL training benchmark
Mirror the runtime benchmark option so sync/async camera training runs can be measured with scripts/benchmarks/training.py directly.
1 parent 83da64b commit 704eac1

2 files changed

Lines changed: 14 additions & 1 deletion

File tree

scripts/benchmarks/rsl_rl/benchmark_rsl_rl_train.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,15 @@ def _parse_args(argv: list[str]):
7373
default=False,
7474
help="Omit per-iteration series data from the bundle to reduce file size.",
7575
)
76+
parser.add_argument(
77+
"--ovrtx_async_rendering",
78+
action=argparse.BooleanOptionalAction,
79+
default=None,
80+
help=(
81+
"Enable or disable OVRTX asynchronous rendering (one frame of camera latency). Requires"
82+
" renderer=ovrtx. When omitted, the task's renderer configuration is left unchanged."
83+
),
84+
)
7685

7786
from scripts.benchmarks.early_stop import add_success_cli_args
7887

@@ -147,6 +156,9 @@ def run(argv: list[str]) -> None:
147156
app_t1 = time.perf_counter_ns()
148157

149158
apply_env_overrides(args_cli, env_cfg)
159+
from scripts.benchmarks.runtime import _apply_ovrtx_async_rendering
160+
161+
ovrtx_async_rendering = _apply_ovrtx_async_rendering(env_cfg, args_cli.ovrtx_async_rendering)
150162
agent_cfg = cli_args.update_rsl_rl_cfg(agent_cfg, args_cli)
151163
agent_cfg.max_iterations = (
152164
args_cli.max_iterations if args_cli.max_iterations is not None else agent_cfg.max_iterations
@@ -172,6 +184,7 @@ def run(argv: list[str]) -> None:
172184
{"name": "seed", "data": agent_cfg.seed},
173185
{"name": "num_envs", "data": env_cfg.scene.num_envs},
174186
{"name": "max_iterations", "data": agent_cfg.max_iterations},
187+
{"name": "ovrtx_async_rendering", "data": ovrtx_async_rendering},
175188
{"name": "presets", "data": ",".join(cfg.presets)},
176189
]
177190
},

source/isaaclab/changelog.d/pv-runtime-benchmark-warmup.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ Added
22
^^^^^
33

44
* Added untimed warmup frames and an ``--ovrtx_async_rendering`` option to the runtime
5-
benchmark.
5+
benchmark, and an ``--ovrtx_async_rendering`` option to the RSL-RL training benchmark.

0 commit comments

Comments
 (0)