Skip to content

fix(rccl/gin): conn-check production nits (AICOMRCCL-2218) - #11822

Merged
dlamd1dai merged 4 commits into
developfrom
users/dondai/aicomrccl-2218-conn-check-nits
Sep 30, 2026
Merged

dlamd1dai merged 4 commits into
developfrom
users/dondai/aicomrccl-2218-conn-check-nits

Conversation

@dlamd1dai

@dlamd1dai dlamd1dai commented Sep 18, 2026 •

Copy link
Copy Markdown
Contributor

Summary

Follow-up to ROCm/rocm-systems#11092 (AICOMRCCL-2217), now rebased onto develop after that PR merged. Production-only conn-check behaviour hardening:

  • Split kMaxConnCheckKernelRanks (1024) from kMaxConnCheckHostRanks (IPC 16) in gin_anvil_conn_check.h with static_assert
  • Null-check conn-check launcher inputs before HIP launch
  • Collapse duplicated step-failure WARNs behind a warnStep lambda
  • Distinguish local vs peer-reported connectivity failures in retry/final diagnostics (global missing is summed over the LSA team)
  • Fold SetConnCheckVerifyMissing into SetConnCheckMissingCalls (negative = always missing)
  • Argus follow-ups: assert pre-launch rejects leave hipSuccess; collapse peer/global retry WARN; clarify stub comment for zero
  • Step-failure WARN names the failed step and does not print the allgather abort sentinel as a missing count
  • Reject an out-of-range selfRank before ginAnvilConnWrite launches

Comm-reuse erase is not here: ROCm/rocm-systems#12218 (AICOMRCCL-2455).

Test plan

  • rccl-UnitTestsGinAnvilPlugin --gtest_filter='*ConnCheck*' — passed 15/15 on smci350-rck-g03-d07-21 (MI350X, gfx950)
  • rccl-UnitTestsGinAnvilConnCheckDevice (null-pointer + rank-cap accept/reject + selfRank reject + no HIP error on reject) — passed 6/6 on smci350-rck-g03-d07-21 (MI350X, gfx950)

JIRA ID: AICOMRCCL-2218

@amd-argus amd-argus Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

ARGUS

Bottom line: LGTM once RejectsTooManyRanksBeforeLaunch also asserts no launch happened. That one is mechanical, so it does not need another pass from me.

Comment @argus review to re-run, or @argus re-review after pushing.

@dlamd1dai dlamd1dai self-assigned this Sep 18, 2026
dlamd1dai added a commit that referenced this pull request Sep 22, 2026
Address Argus nits on #11822 and move the conn-checked set erase from
unreachable plugin finalize into closeColl so a recycled ncclComm* cannot
skip the gate after HostFinalize.

Co-authored-by: Cursor <cursoragent@cursor.com>
@dlamd1dai

Copy link
Copy Markdown
Contributor Author

Pushed acd858655f5:

  1. Argus fix(rccl/gin): conn-check production nits (AICOMRCCL-2218) #11822 nits: pre-launch rank/null rejects clear/peek HIP error; peer vs global retry WARN collapse; stub comment documents zero = never.
  2. Dedup-set erase moved from ginAnvilFinalize → ginAnvilCloseColl, with ConnCheck_CloseCollUnmarksCommForReuse (closeColl without finalize, second bind must re-run the gate). Addresses the recycled-ncclComm* skip that Argus flagged on fix(rccl/gin-sdma): restore LSA conn-check with device-linked TU (AICOMRCCL-2217) #11092; keeping that fix off fix(rccl/gin-sdma): restore LSA conn-check with device-linked TU (AICOMRCCL-2217) #11092 per stack split.

Still blocked on #11092. Will rebase onto develop after that lands.

@amd-argus amd-argus Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

ARGUS

Bottom line: LGTM once the step-failure case stops printing the lsaTeamSize sentinel as local missing at gin_plugin_anvil_sdma.cc:699. That one is mechanical, so it does not need another pass from me.

Comment @argus review to re-run, or @argus re-review after pushing.

@dlamd1dai

Copy link
Copy Markdown
Contributor Author

Comm-reuse closeColl erase + reuse gtest moved to #12218 (AICOMRCCL-2455) so this PR stays 2218 production nits only.

@dlamd1dai
dlamd1dai force-pushed the users/dondai/aicomrccl-2217-conn-check-restore branch from c21fcd0 to 4392d1c Compare September 24, 2026 18:24
dlamd1dai added a commit that referenced this pull request Sep 24, 2026
Address Argus nits on #11822 and move the conn-checked set erase from
unreachable plugin finalize into closeColl so a recycled ncclComm* cannot
skip the gate after HostFinalize.

Co-authored-by: Cursor <cursoragent@cursor.com>
@dlamd1dai
dlamd1dai force-pushed the users/dondai/aicomrccl-2218-conn-check-nits branch from 137181b to bd12427 Compare September 24, 2026 19:29

@amd-argus amd-argus Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

ARGUS

Bottom line: Worth settling before approval: the gate-failure WARN at gin_plugin_anvil_sdma.cc:694 still prints the lsaTeamSize sentinel as a measured local missing count.

Comment @argus review to re-run, or @argus re-review after pushing.

dlamd1dai and others added 3 commits September 28, 2026 15:15
Split device vs host rank caps in gin_anvil_conn_check.h, null-check
conn-check launch inputs, fold verify-missing into the missing-call
counter, and tighten step/peer failure diagnostics.

Stacked on users/dondai/aicomrccl-2217-conn-check-restore (#11092).

Co-authored-by: Cursor <cursoragent@cursor.com>
Address Argus nits on #11822 and move the conn-checked set erase from
unreachable plugin finalize into closeColl so a recycled ncclComm* cannot
skip the gate after HostFinalize.

Co-authored-by: Cursor <cursoragent@cursor.com>
Comm-reuse un-mark lives on stacked PR for AICOMRCCL-2455 so this PR
stays production nits only.
@dlamd1dai
dlamd1dai force-pushed the users/dondai/aicomrccl-2218-conn-check-nits branch from bd12427 to adc0aa1 Compare September 28, 2026 19:16
…inel

A local step failure sets localMissing to the team size only so the allgather aborts every rank. Logging that value as a measured missing count hides the failed step.

Co-authored-by: Cursor <cursoragent@cursor.com>

@amd-argus amd-argus Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

ARGUS

Bottom line: LGTM.

Comment @argus review to re-run, or @argus re-review after pushing.

@abouteiller abouteiller left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

@systems-assistant

Copy link
Copy Markdown
Contributor

TheRock Submodule Bump Activity

Newest first

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants