Skip to content

[DO NOT COMMIT][di] D2D transfer#31

Draft
aarondou wants to merge 1 commit intov0.11.0from
di-poc-d2d
Draft

[DO NOT COMMIT][di] D2D transfer#31
aarondou wants to merge 1 commit intov0.11.0from
di-poc-d2d

Conversation

@aarondou
Copy link

@aarondou aarondou commented Nov 7, 2025

Essential Elements of an Effective PR Description Checklist

  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.

Purpose

Test Plan

Test Result

(Optional) Documentation Update

caches_data = []
# With hybrid allocator, layers can share a kv cache tensor
seen_base_addresses = []
seen_base_addresses_dict = {}
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can be removed with the following API
int nrt_get_device_index(void *tensor_va); // returns device rank

@aarondou aarondou changed the title [di] D2D transfer [DO NOT COMMIT][di] D2D transfer Nov 7, 2025
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