-
Notifications
You must be signed in to change notification settings - Fork 825
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Token seq id #5964
Token seq id #5964
Conversation
…ata_transport_token_per_placement
…ata_transport_token_per_placement
placement = flow.placement("cuda", {0: range(2)}) | ||
sbp = (flow.sbp.split(0),) | ||
for i in range(1000): | ||
y = x.to_consistent(placement=placement, sbp=sbp) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这会在main线程上同步元信息。
sbp = (flow.sbp.split(0),) | ||
for i in range(1000): | ||
y = x.to_consistent(placement=placement, sbp=sbp) | ||
b = a.to_consistent(placement=placement, sbp=flow.sbp.broadcast) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这会在scheduler县城上同步张量数据。
CI failed, removing label automerge |
CI failed, removing label automerge |
CI failed, removing label automerge |
1 similar comment
CI failed, removing label automerge |
Speed stats:
|
不再固定token传输main线程上的数据。而是会为每个token在实际传输的时候附带序列号。
序列号在(src_token, dst_token, thread_consistent_id, rank_group)下分类独立自增。