Skip to content

CFD UQ Integration [2/2]: Add member/domain parallel example - #53

Open
ktangsali wants to merge 1 commit into
codex/cfd-uq-01-statisticsfrom
codex/cfd-uq-02-distributed-example
Open

ktangsali wants to merge 1 commit into
codex/cfd-uq-01-statisticsfrom
codex/cfd-uq-02-distributed-example

Conversation

@ktangsali

@ktangsali ktangsali commented Nov 12, 2026 •

Copy link
Copy Markdown
Collaborator

Show how CFD workflows can reduce ensemble members while retaining spatial partitions, then feed final summaries into existing metrics. The executable example checks GP total variance, a pressure-force observable, unequal member counts, uneven spatial partitions, an empty final contribution, and isolation between independent cases.

Validation: all three CPU/Gloo layouts passed on PyTorch 2.15 (two ranks with member-only or domain-only parallelism, and four ranks combining both). Pre-commit passed. GPU validation: all three layouts also passed with NCCL on H200, including the four-GPU 2-member-group × 2-domain-partition case (NRT job 844008, commit 8c4f385). The full CFD suite passed all 197 tests. Integrated GP uncertainty still requires joint samples or cross-point covariance.

Second of two stacked PRs; targets codex/cfd-uq-01-statistics so the diff contains only the example, its documentation, and CI execution.

Before merging (after #52 merges):

  • Retarget/rebase this PR onto main.
  • Remove "codex/cfd-uq-*" and its explanatory comments from .github/workflows/ci-tests.yml, restoring branches: [main, master]. Keep the filter while this PR still targets CFD UQ Integration [1/2]: Add shared streaming statistics backend #52's branch so CI continues to run.
  • Confirm CI passes against main after the cleanup, including the distributed layouts.

@ktangsali
ktangsali marked this pull request as ready for review November 12, 2026 21:36
Signed-off-by: Kaustubh Tangsali <ktangsali@nvidia.com>
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