8000 [a2av] Align length of major dimension in output of 2D a2av by kwen2501 · Pull Request #155172 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[a2av] Align length of major dimension in output of 2D a2av #155172

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.

Sign up for GitHub

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

Closed
kwen2501 wants to merge 10 commits into from
Closed

[a2av] Align length of major dimension in output of 2D a2av #155172

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
79094e2
Update
kwen2501 Jun 4, 2025
ecfeaf4
Update
kwen2501 Jun 4, 2025
754d6bd
Update
kwen2501 Jun 5, 2025
93041d6
Update
kwen2501 Jun 5, 2025
b8db459
Update
kwen2501 Jun 5, 2025
8a889c0
Update
kwen2501 Jun 5, 2025
ba6b78d
Update
kwen2501 Jun 5, 2025
75cc801
Update
kwen2501 Jun 5, 2025
741b5a8
Update
kwen2501 Jun 5, 2025
fd596f4
Update
kwen2501 Jun 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update 
[ghstack-poisoned]
  • Loading branch information
@kwen2501
kwen2501 committed Jun 5, 2025
commit 754d6bdda1a21cc3ba2d5be7839a3df8e82542ba
2 changes: 1 addition & 1 deletion torch/csrc/distributed/c10d/SymmetricMemory.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -281,7 +281,7 @@ TORCH_LIBRARY_FRAGMENT(symm_mem, m) {
m.def(
"nvshmem_all_to_all_vdev(Tensor input, Tensor(a!) out, Tensor(a!) in_out_splits, str group_name) -> Tensor(a!)");
m.def(
"nvshmem_all_to_all_vdev_2d(Tensor input, Tensor(a!) out, Tensor(a!) in_out_splits, str group_name, int majoy_align) -> Tensor(a!)");
"nvshmem_all_to_all_vdev_2d(Tensor input, Tensor(a!) out, Tensor(a!) in_out_splits, str group_name, int major_align) -> Tensor(a!)");
}

TORCH_LIBRARY_IMPL(symm_mem, Meta, m) {
Expand Down
Loading
0