Skip to content

gin: Fix regMrSym hang caused by duplicate detection - #1372

Open
rauteric wants to merge 1 commit into
aws:masterfrom
rauteric:fix/gin-symm-mr-dedup-allgather
Open

gin: Fix regMrSym hang caused by duplicate detection#1372
rauteric wants to merge 1 commit into
aws:masterfrom
rauteric:fix/gin-symm-mr-dedup-allgather

Conversation

@rauteric

Copy link
Copy Markdown
Contributor

The plugin maintains a refcounted, address-keyed dedup cache for symmetric registrations. In case of a duplicate registration, the previous code used the existing registration and exited without participating in the all_gather, causing a hang if some ranks went through the normal registration process. (This is possible in particular if some -- but not all -- ranks deregistered the region before reregistering it.)

Fix: all ranks will now go through the all_gather after local registration, even if it is a duplicate registration. Also added a functional test for this case.

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

@rauteric
rauteric requested a review from a team as a code owner August 24, 2026 23:53
The plugin maintains a refcounted, address-keyed dedup cache for symmetric
registrations.  In case of a duplicate registration, the previous code used the
existing registration and exited without participating in the all_gather,
causing a hang if some ranks went through the normal registration process.
(This is possible in particular if some -- but not all -- ranks deregistered
the region before reregistering it.)

Fix: all ranks will now go through the all_gather after local registration,
even if it is a duplicate registration. Also added a functional test for this
case.

Signed-off-by: Eric Raut <eraut@amazon.com>
@rauteric
rauteric force-pushed the fix/gin-symm-mr-dedup-allgather branch from 30ad7a7 to 3da6dab Compare August 24, 2026 23:57
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