do not submit

Signed-off-by: Qiliang Cui <[email protected]>

Author: QiliangCui

.buildkite/pipeline_jax.yml

@@ -8,6 +8,19 @@ steps:
   # -----------------------------------------------------------------
   # TEST STEPS - Calling wrapper
   # -----------------------------------------------------------------
+
+   - label: "Debug Git Diff"
+     key: debug_diff
+     agents:
+      queue: tpu_v6e_queue
+     commands:
+       - echo "--- Git Commit Info"
+       - git show --summary
+       - echo "--- Git Diff (HEAD~1)"
+       - git diff --name-only HEAD~1
+       - echo "--- Testing Glob Match"
+       - git diff --name-only HEAD~1 | grep "tpu_inference/kernels/" || echo "No kernel match found via grep"
+
    - label: "E2E MLPerf tests for JAX models"
      key: test_0
      soft_fail: true

tpu_inference/kernels/ragged_paged_attention/v3/kernel.py

@@ -23,6 +23,7 @@
 DEFAULT_VMEM_LIMIT_BYTES = 100 * 1024 * 1024
 
 
+# trigger a test
 def ref_ragged_paged_attention(
     queries: jax.
     Array,  # [max_num_tokens, actual_num_q_heads, actual_head_dim]