feat(runtime): Add offset device copy wrapper - #2164
Conversation
|
Important Review skippedDraft detected. Please check the settings in the CodeRabbit UI or the ⚙️ Run configurationConfiguration used: Organization UI Review profile: CHILL Plan: Pro Plus Run ID: You can disable this status message by setting the Use the checkbox below for a quick retry:
Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out. Comment |
|
Closing as superseded by the revised simpler #1538, which fixes range validation in the existing |
Summary
DistributedWorker.copy_to_offsetfor H2D copies into an offset of a live device allocation.DistributedWorkerand cover its argument forwarding with a unit test.Motivation
Simpler validates device-pointer provenance and rejects interior pointers passed through
copy_to.Callers updating a slice of a persistent allocation therefore need an explicit base-plus-offset API.
This wrapper exposes that API to PyPTO and unblocks offset cache-page updates in serving workloads.
Dependency
Validation
cmake --build build --parallelpython -m pytest tests/ut/runtime/test_distributed_worker.py -n auto --maxprocesses 8 -q: 104 passed_task_interfacepredating the simpler source-tree build stamp