Skip to content

bench(env): use tensor actions for env rollouts - #1725

Merged
TATP-233 merged 3 commits into
develop/tensor-runtimefrom
feat/issue-1704-tensor-benchmark-actions
Sep 29, 2026
Merged

TATP-233 merged 3 commits into
develop/tensor-runtimefrom
feat/issue-1704-tensor-benchmark-actions

Conversation

@TATP-233

Copy link
Copy Markdown
Collaborator

Summary

  • Allocate benchmark_env_step warmup and measured actions as contiguous float32 Torch tensors on env.device.
  • Replace the SimToolReal fixed-tool benchmark NumPy action carrier with a device-resident Torch tensor.
  • Validate SimToolReal rollout finiteness with Torch while retaining one explicit scalar publication to the benchmark result.
  • Add a focused contract test preventing the env-step benchmark from regressing to np.random.uniform actions.

Part of #1704.

Validation

  • UNILAB_LOCAL_UNISIM=/home/user/ws/unilabsim/unisim uv run pytest -q tests/benchmark/test_env_step_config_contract.py — 7 passed
  • UNILAB_LOCAL_UNISIM=/home/user/ws/unilabsim/unisim uv run make check
  • UNILAB_LOCAL_UNISIM=/home/user/ws/unilabsim/unisim uv run make test-all — 1833 passed, 26 skipped; benchmark imports 35/35 and 36/36

Remaining #1704 benchmark work

The phase-CPU benchmark private-API probe, NpEnv shard benchmark replacement, off-policy active benchmark tensor transition, and neutral timing metric rename remain open.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant