Remove the grid override to bypass OOM, Add the XLA vmodule flag to print verbose log for partitioner.

Author: felixwqp

.github/workflows/test-gb-25.yml

@@ -189,11 +189,8 @@ jobs:
       - name: Run GB-25 simulation
         timeout-minutes: 60
         run: |
-          export GB25_GRID_X="6144"
-          export GB25_GRID_Y="1536"
-          export GB25_GRID_Z="4"
           export XLA_FLAGS='--xla_dump_to=${{ env.GB25_DIR }}/xla_dump'
-          timeout --signal=TERM --verbose 59m mpiexecjl -np 1 julia --color=yes --project -O0 sharding/sharded_baroclinic_instability_simulation_run.jl
+          timeout --signal=TERM --verbose 59m mpiexecjl -np 1 julia --vmodule=spmd_partitioner*=10 --color=yes --project -O0 sharding/sharded_baroclinic_instability_simulation_run.jl
         working-directory: ${{ env.GB25_DIR }}
       - name: Test correctness in GB-25 code
         timeout-minutes: 20