Skip to content

chore(deps): consume published tensor runtime packages - #2071

Merged
TATP-233 merged 656 commits into
mainfrom
develop/tensor-runtime
Oct 8, 2026
Merged

TATP-233 merged 656 commits into
mainfrom
develop/tensor-runtime

Conversation

@TATP-233

@TATP-233 TATP-233 commented Oct 8, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Removes the temporary tensor-runtime sibling workspace scaffolding: tensor_runtime_workspace.json, its materializer script, tests, Make targets, CI materialization steps, and UNILAB_LOCAL_UNISIM overrides.
  • Switches default and ROCm dependency profiles from relative editable sibling checkouts to published registry packages:
    • unisim-core>=1.7.11
    • unilab-rl==1.4.5
    • mjbatch-uni~=0.2.5
  • Updates lock files for both profiles and refreshes English/Chinese workflow and production documentation.
  • Adapts off-policy dispatch to unilab-rl 1.4.5, which folded the age-biased WarpSAC runtime into FlashSAC. UniLab retains the public warpsac config and entrypoint, dispatching to the maintained FlashSAC owner implementation.
  • Makes the MPS pre-environment test deterministic with a local control socket rather than relying on the host /tmp/nvidia-mps daemon state.

Linked Work

  • Issue: tensor-runtime published dependency transition / M11 closeout
  • Parent roadmap: tensor runtime
  • Roadmap declared base: main
  • Milestone: tensor runtime productionization
  • Base branch: main

Validation

  • make test-all passed on the final local head before this PR was created or updated
  • Additional task-specific validation listed below

Commands actually run:

uv lock --check
uv run pytest -q \
  tests/base/test_backend_imports.py \
  tests/training/test_soak.py \
  tests/scripts/test_g1_flashsac_backend.py \
  tests/scripts/test_rocm_dependency_profile.py \
  tests/algos/test_offpolicy_double_buffer_runner.py \
  tests/training/test_cuda_process_sharing.py \
  tests/tasks/test_tensor_runtime_components.py \
  tests/tasks/test_go2_tensor_runtime.py
# 187 passed, 1 skipped

make check
# passed

make test
# 1862 passed, 50 skipped, 551 deselected

make test-cov
# passed

uv run python scripts/benchmark/smoke_test.py
# passed

Remote CI route:

  • Base main: current-head CI link and status to be recorded after PR creation.

Impact

  • Backend impact: both
  • Platform impact: both
  • Training effect expected: yes

Training now consumes published unisim-core, unilab-rl, and mjbatch-uni versions. WarpSAC keeps its UniLab task/config identity but dispatches through the FlashSAC runtime owner after the upstream algorithm merge.

Artifacts

  • W&B: not applicable
  • benchmark result: not applicable
  • video / screenshot: not applicable
  • ONNX / checkpoint: not applicable

Checklist

  • Added or updated tests where needed
  • Updated docs if behavior or workflow changed
  • Linked the driving issue
  • Noted any follow-up work explicitly

TATP-233 and others added 30 commits October 2, 2026 19:23
…chmark

test(benchmark): scope flashsac backend owners
feat(config): remove shelved task owners
refactor(obs): tensor-native delay state
refactor(obs): tensor-native noise models
perf(command): publish uniform velocity tensor
perf(command): keep G1 resampling on device
…ocol

feat(managers): add reset-row observation protocol
feat(reset): publish call-graph diagnostics
feat(motion): add tensor-native sampler
feat(managers): declare reset owner contract
…ries

feat(managers): expose reset state boundaries
feat(reset): execute reset owner lifecycle
perf(motion): own command action metric reset
TATP-233 and others added 21 commits October 4, 2026 14:37
Issue #2054. One-line test-scope correction; focused MuJoCo/Motrix slow owner tests pass.
Issue #2054. Non-canonical Motrix cells are MISSING rather than absent. Focused slow support-matrix tests pass.
Closes #2054. Integrates canonical Motrix owners, packed HOST_BRIDGE contract hardening, selected-row read optimization, docs/support/CI updates, and measured local performance gates into develop/tensor-runtime.
…time

feat(macos): enable mujoco/mjbatch tensor-runtime off-policy training on macOS
chore(workspace): bump unilab-rl pin for terminal reward fix
Remove the G1FlipTracking and G1FlipTrackingSAC registry identities, their
ppo/appo/sac owner YAMLs, the associated profile, partial-reset, owner
contract, and smoke test coverage, support-matrix entries, and the
motion-tracking and sim2sim docs references. x2_wall_flip_tracking and the
generic flip CSV motion tooling are unaffected.
refactor(tasks): remove g1_flip_tracking task
feat(training): add single-rank CUDA MPS mode
feat(drake): scope Go2 owner into tensor runtime
feat(training): enable single-host multi-GPU MPS
feat(superdex): restore TorchEnv CPU host-bridge owners
@TATP-233
TATP-233 force-pushed the develop/tensor-runtime branch from 0fcf4a5 to 6a2e59e Compare October 8, 2026 08:34
@TATP-233

TATP-233 commented Oct 8, 2026

Copy link
Copy Markdown
Collaborator Author

Remote CI follow-ups on the current head:

  • Fixed the CPU pyright diagnostics access to optional torch.version metadata (6a2e59ea).
  • Bumped CPU-only CI torch installations to 2.10.0 (667158b6). The prior 2.7/2.8 CPU wheels still crash when unisim-core clears the CPU default device context, which is the contract exercised by unisim-core 1.7.11.

Current-head CI rerun: https://github.com/Motphys/UniLab/actions/runs/37750794426 (superseded); latest run is being recorded after 667158b.

@TATP-233

TATP-233 commented Oct 8, 2026

Copy link
Copy Markdown
Collaborator Author

Current-head remote CI is green:

@TATP-233
TATP-233 merged commit 1ee4709 into main Oct 8, 2026
8 checks passed
@TATP-233
TATP-233 deleted the develop/tensor-runtime branch October 8, 2026 09:03
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