Record CUDA JAX throughput
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user