Record CUDA JAX throughput

2026-07-04 21:56:24 +09:00
parent 72893250ec
commit b37b841eef
1 changed files with 14 additions and 0 deletions
@@ -158,6 +158,9 @@ Throughput benchmark:

 ```bash
 flock -n .compute.lock uv run python benchmarks/throughput.py
+
+# Optional CUDA check without making CUDA a project dependency:
+flock -n .compute.lock uv run --with 'jax[cuda12]' python benchmarks/throughput.py
 ```

 Measured on 2026-07-04 with CPU JAX backend:
@@ -170,6 +173,17 @@ elapsed_sec=4.655526
 steps_per_sec=450465.14
 ```

+Measured on 2026-07-04 with CUDA JAX backend on RTX 3090, using the optional
+`uv run --with 'jax[cuda12]' ...` command:
+
+```text
+backend=gpu
+batch_size=8192
+steps=256
+elapsed_sec=0.530039
+steps_per_sec=3956598.38
+```
+
 ### DECISIONS.md

 - Explicit `deck_order` dealing uses the first eight cards for player 0 and