SIGN IN SIGN UP

[Pallas/GPU interpreter] Relax test tolerances and use NumPy matmul for reference.

Tests failed in macOS CI due to small numerical differences between NumPy and JAX matrix multiplication outputs.

* Relax tolerances to 1e-6 in `gpu_pallas_interpret_tcgen05_race_detection_test.py`.
* In `gpu_pallas_interpret_test.py`, compute the expected array with `np.matmul` to match the interpreter's implementation and keep `assertArraysEqual`.

PiperOrigin-RevId: 972632869
P
Peter Hawkins committed
b277e8edcce846a2f14b943f6893ce4aac3ae2c3
Parent: 7d2846c
Committed by jax authors <google-ml-automation@google.com> on 8/28/2026, 4:31:52 PM