From 6298255ebd8aef1e541903328c849ff11849dc22 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 13:21:02 -0500 Subject: [PATCH 1/6] Test dus_to_pad within xla --- .github/workflows/test-gb-25.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index d9680e83b..1cd83ded0 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -48,13 +48,13 @@ jobs: - '1.11' xla_commit: - '' - # - 'b25f3cbed2bc88c8ffef85f6a5319e2cf7b0454c' + - '7e740cc0c4b5d6e43e7e35f7f9f07c6152b62999' gb25_commit: - 'main' # - '0123456789abcdef0123456789abcdef01234567' reactant_commit: - 'main' - # - 'regenerate-mlir-bindings' + - 'wsmoses-patch-9' steps: - name: Check GPUs From c55318d739d91c1e6ad4e340aeaec43739914b5b Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 15:12:21 -0500 Subject: [PATCH 2/6] Update test-gb-25.yml --- .github/workflows/test-gb-25.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index 1cd83ded0..667f5a5ea 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -48,7 +48,7 @@ jobs: - '1.11' xla_commit: - '' - - '7e740cc0c4b5d6e43e7e35f7f9f07c6152b62999' + - 'dd0f59f000f23641191db8b2a1f8f893e429cca4' gb25_commit: - 'main' # - '0123456789abcdef0123456789abcdef01234567' From bee4d9dcd73a2e60be2b677087e0107a1472b119 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 19:52:46 -0500 Subject: [PATCH 3/6] Add GPU check step to workflow --- .github/workflows/test-gb-25.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index 667f5a5ea..afb9ae8d0 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -56,6 +56,7 @@ jobs: - 'main' - 'wsmoses-patch-9' + steps: - name: Check GPUs if: ${{ contains(matrix.os, 'a100') }} From b323ab2c703fee165fc28865a629c7a8e872c49d Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 20:58:37 -0500 Subject: [PATCH 4/6] Update test-gb-25.yml --- .github/workflows/test-gb-25.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index afb9ae8d0..7a79e0531 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -50,7 +50,8 @@ jobs: - '' - 'dd0f59f000f23641191db8b2a1f8f893e429cca4' gb25_commit: - - 'main' + # - 'main' + - 'mg/oom-reproducer' # - '0123456789abcdef0123456789abcdef01234567' reactant_commit: - 'main' From c806a3ebd98b0c1c7a0c4221ff1e8de765086cd6 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 21:44:17 -0500 Subject: [PATCH 5/6] Update test-gb-25.yml --- .github/workflows/test-gb-25.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index 7a79e0531..6814177e8 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -189,7 +189,7 @@ jobs: - name: Run GB-25 simulation timeout-minutes: 60 run: | - export XLA_FLAGS='--xla_dump_to=${{ env.GB25_DIR }}/xla_dump' + export XLA_FLAGS='--xla_dump_to=${{ env.GB25_DIR }}/xla_dump --xla_dump_hlo_pass_re=.*' timeout --signal=TERM --verbose 59m mpiexecjl -np 1 julia --color=yes --project -O0 --startup-file=no --threads=16 --compiled-modules=strict sharding/sharded_baroclinic_instability_simulation_run.jl working-directory: ${{ env.GB25_DIR }} - name: Test correctness in GB-25 code From a1969b1a40349df90a64593cf120b5a687526409 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sun, 2 Nov 2025 10:34:59 -0600 Subject: [PATCH 6/6] Update test-gb-25.yml --- .github/workflows/test-gb-25.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test-gb-25.yml b/.github/workflows/test-gb-25.yml index 6814177e8..69bbb8f3c 100644 --- a/.github/workflows/test-gb-25.yml +++ b/.github/workflows/test-gb-25.yml @@ -50,8 +50,8 @@ jobs: - '' - 'dd0f59f000f23641191db8b2a1f8f893e429cca4' gb25_commit: - # - 'main' - - 'mg/oom-reproducer' + - 'main' + # - 'mg/oom-reproducer' # - '0123456789abcdef0123456789abcdef01234567' reactant_commit: - 'main'