fix: charge the prefill logits GEMM for MoE-family models

.github/actionlint.yaml

@@ -6,5 +6,6 @@
 # them.
 self-hosted-runner:
   labels:
-    - prod-aisimulate-default-amd64-v1
-    - prod-aisimulate-default-arm64-v1
+    - prod-aisim-default-amd64-v1
+    - prod-aisim-default-arm64-v1
+    - dynamo-cpu-4gi-amd64

.github/workflows/ci.yml

@@ -268,19 +268,19 @@ jobs:
       matrix:
         include:
           - arch: amd64
-            runner: prod-aisimulate-default-amd64-v1
+            runner: prod-aisim-default-amd64-v1
             cargo_deny_target: x86_64-unknown-linux-musl
             cargo_deny_sha256: 0e8c2aa59128612c90d9e09c02204e912f29a5b8d9a64671b94608cbe09e064f
           - arch: arm64
-            runner: prod-aisimulate-default-arm64-v1
+            runner: prod-aisim-default-arm64-v1
             cargo_deny_target: aarch64-unknown-linux-musl
             cargo_deny_sha256: 2b3567a60b7491c159d1cef8b7d8479d1ad2a31e29ef49462634ad4552fcc77d
     runs-on: ${{ matrix.runner }}
     timeout-minutes: 10
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -317,11 +317,11 @@ jobs:
       fail-fast: false
       matrix:
         arch: [amd64, arm64]
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -336,14 +336,14 @@ jobs:
     name: Rust feature modes
     needs: [fast-ci, select-full-ci]
     if: ${{ needs.select-full-ci.outputs.rust_feature_modes == 'true' }}
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 15
     env:
       CARGO_TARGET_DIR: ${{ github.workspace }}/target-feature-tests
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -382,11 +382,11 @@ jobs:
       fail-fast: false
       matrix:
         arch: [amd64, arm64]
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -413,11 +413,11 @@ jobs:
       fail-fast: false
       matrix:
         arch: [amd64, arm64]
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -462,11 +462,11 @@ jobs:
       fail-fast: false
       matrix:
         arch: [amd64, arm64]
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c # v8.0.1
@@ -601,12 +601,12 @@ jobs:
           - arch: arm64
             # quay.io/pypa/manylinux_2_28_aarch64:2026.08.24-1
             container_image: quay.io/pypa/manylinux_2_28_aarch64@sha256:29a6ceb78de5b00494e6e7456a72782a4cb54238de102e7491d15fe47789b4d1
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     timeout-minutes: 10
     container:
       image: ${{ matrix.container_image }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -644,12 +644,12 @@ jobs:
           - {suite: cli-build, group: 4, groups: 4}
           - {suite: support-matrix, group: 1, groups: 1}
           - {suite: tools-build, group: 1, groups: 1}
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     timeout-minutes: 15
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -765,12 +765,12 @@ jobs:
         needs.select-full-ci.result == 'success' &&
         needs.select-full-ci.outputs.application_wheel == 'true'
       }}
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 15
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -787,7 +787,7 @@ jobs:
     name: Python ${{ matrix.python-version }} compatibility
     needs: [fast-ci, select-full-ci]
     if: ${{ needs.select-full-ci.outputs.python_compatibility == 'true' }}
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 30
     strategy:
       fail-fast: false
@@ -796,7 +796,7 @@ jobs:
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -820,12 +820,12 @@ jobs:
     name: Engine Golden Regression
     needs: [fast-ci, select-full-ci]
     if: ${{ needs.select-full-ci.outputs.engine_golden_regression == 'true' }}
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 45
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -873,11 +873,11 @@ jobs:
           - arch: arm64
             # quay.io/pypa/manylinux_2_28_aarch64:2026.08.24-1
             container_image: quay.io/pypa/manylinux_2_28_aarch64@sha256:29a6ceb78de5b00494e6e7456a72782a4cb54238de102e7491d15fe47789b4d1
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     container:
       image: ${{ matrix.container_image }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7

.github/workflows/collector-check.yml

@@ -31,12 +31,12 @@ concurrency:
 jobs:
   check:
     name: Check collector data
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 20
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - name: Checkout code
@@ -116,13 +116,13 @@ jobs:
 
   cross-backend-consistency:
     name: Perf data sanity (informational)
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 25
     continue-on-error: true
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - name: Checkout code

.github/workflows/e2e-accuracy-branch.yml

@@ -34,7 +34,7 @@ permissions:
 jobs:
   wheel:
     name: Prepare exact accuracy wheel
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 120
     permissions:
       contents: read
@@ -123,11 +123,11 @@ jobs:
   measurements:
     name: Prepare shared E2E measurements
     timeout-minutes: 120
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7
@@ -176,13 +176,13 @@ jobs:
       max-parallel: 4
       matrix:
         shard: [0, 1, 2, 3]
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 360
     container:
       # The runner executes as non-root; use its sudo-enabled CI image.
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     env:
       OPENBLAS_NUM_THREADS: '1'
@@ -286,11 +286,11 @@ jobs:
     name: Qualify E2E accuracy (${{ inputs.artifact_key }})
     needs: [wheel, shards]
     timeout-minutes: 30
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7

.github/workflows/fpe-release-qualify.yml

@@ -26,7 +26,7 @@ jobs:
   prepare:
     name: Build and identify release FPE wheel
     if: github.ref == 'refs/heads/main'
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 30
     container:
       # Same pinned manylinux toolchain as the main FPE producer.
@@ -74,7 +74,7 @@ jobs:
   generate:
     name: Release FPE (${{ matrix.shard.system }}, ${{ matrix.shard.backend }})
     needs: prepare
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 480
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}

.github/workflows/fpe-support-matrix.yml

@@ -35,13 +35,13 @@ env:
 jobs:
   prepare-wheel:
     name: Prepare exact FPE wheel
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 30
     container:
       # quay.io/pypa/manylinux_2_28_x86_64:2026.08.24-1
       image: quay.io/pypa/manylinux_2_28_x86_64@sha256:4dc41da7df20400310c80d162a2fe2d2c2f3d9734d8dec20f6b9843711618deb
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     outputs:
       wheel-sha256: ${{ steps.digest.outputs.sha256 }}
@@ -148,12 +148,12 @@ jobs:
   generate:
     name: FPE matrix (${{ matrix.shard.system }}, ${{ matrix.shard.backend }})
     needs: [discover-shards, prepare-wheel]
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 480
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     strategy:
       fail-fast: false

.github/workflows/fpm-accuracy-branch.yml

@@ -29,7 +29,7 @@ permissions:
 jobs:
   wheel:
     name: Prepare exact accuracy wheel
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 120
     permissions:
       contents: read
@@ -118,7 +118,7 @@ jobs:
   campaign:
     name: Qualify FPM accuracy (${{ inputs.artifact_key }})
     needs: wheel
-    runs-on: prod-aisimulate-default-amd64-v1
+    runs-on: prod-aisim-default-amd64-v1
     timeout-minutes: 360
     container:
       image: quay.io/pypa/manylinux_2_28_x86_64@sha256:4dc41da7df20400310c80d162a2fe2d2c2f3d9734d8dec20f6b9843711618deb

.github/workflows/nightly-ci.yml

@@ -292,12 +292,12 @@ jobs:
       matrix:
         arch: [amd64, arm64]
         python-version: ["3.11", "3.12", "3.13"]
-    runs-on: prod-aisimulate-default-${{ matrix.arch }}-v1
+    runs-on: prod-aisim-default-${{ matrix.arch }}-v1
     timeout-minutes: 15
     container:
       image: ${{ vars.CI_JOB_CONTAINER_IMAGE }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
         USER: runner
     steps:
       # Keep the license policy/tooling at the workflow revision. Older
@@ -351,15 +351,15 @@ jobs:
       matrix:
         include:
           - arch: amd64
-            runner: prod-aisimulate-default-amd64-v1
+            runner: prod-aisim-default-amd64-v1
             rust_target: x86_64-unknown-linux-gnu
             rustup_init_sha256: 4acc9acc76d5079515b46346a485974457b5a79893cfb01112423c89aeb5aa10
             uv_target: x86_64-unknown-linux-gnu
             uv_sha256: 8681d8921e7d520fb368991dcf5f9c1905b80f5bf2a265a0ed085c8d8e342477
             # quay.io/pypa/manylinux_2_28_x86_64:2026.08.24-1
             container_image: quay.io/pypa/manylinux_2_28_x86_64@sha256:4dc41da7df20400310c80d162a2fe2d2c2f3d9734d8dec20f6b9843711618deb
           - arch: arm64
-            runner: prod-aisimulate-default-arm64-v1
+            runner: prod-aisim-default-arm64-v1
             rust_target: aarch64-unknown-linux-gnu
             rustup_init_sha256: 9732d6c5e2a098d3521fca8145d826ae0aaa067ef2385ead08e6feac88fa5792
             uv_target: aarch64-unknown-linux-gnu
@@ -379,7 +379,7 @@ jobs:
       # for the runner's node24-based actions.
       image: ${{ matrix.container_image }}
       env:
-        HOME: /__w/aisimulate/aisimulate
+        HOME: /__w/aisim/aisim
     defaults:
       run:
         shell: bash