Commit 726bfe5
authored
Fix lock signal scope for cross-GPU visibility in matmul_all_reduce (#477)
Use scope="sys" instead of scope="gpu" for the lock-signal atomic in
_fused_matmul_all_reduce_kernel. The lock array lives on the symmetric
heap mapped via IPC, so atomics consumed by remote ranks must use
system scope for xGMI visibility. Fixes intermittent hangs on
multi-GPU runs.
Fixes #462
Co-Authored-By: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>1 parent 0f27599 commit 726bfe5
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
132 | 132 | | |
133 | 133 | | |
134 | 134 | | |
135 | | - | |
| 135 | + | |
136 | 136 | | |
137 | 137 | | |
138 | 138 | | |
| |||
0 commit comments