Record CUDA JAX throughput

This commit is contained in:
2026-07-04 21:56:24 +09:00
parent 72893250ec
commit b37b841eef
+14
View File
@@ -158,6 +158,9 @@ Throughput benchmark:
```bash ```bash
flock -n .compute.lock uv run python benchmarks/throughput.py 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: Measured on 2026-07-04 with CPU JAX backend:
@@ -170,6 +173,17 @@ elapsed_sec=4.655526
steps_per_sec=450465.14 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 ### DECISIONS.md
- Explicit `deck_order` dealing uses the first eight cards for player 0 and - Explicit `deck_order` dealing uses the first eight cards for player 0 and