Skip to content

Commit b8fa8d9

Browse files
larsgebclaude
andcommitted
demo: remove 50M-element benchmark size that OOMs the CI paravirtual GPU
The 1D benchmark ran up to N=50_000_000 (200 MB per array, 600 MB for the three Metal buffers simultaneously in flight). The CI runner's paravirtual GPU cannot allocate that much Metal shared memory and crashes with SIGSEGV. N=10_000_000 (40 MB per buffer) already captures the crossover point where GPU wins over NumPy and keeps total memory well within the runner's limits. Also remove the per-cell progress logging that was added for diagnosis. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 5516665 commit b8fa8d9

2 files changed

Lines changed: 0 additions & 5 deletions

File tree

demo.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,6 @@ def benchmark(func, *args, repeats=100, warmup=15):
117117
100_000,
118118
1_000_000,
119119
10_000_000,
120-
50_000_000,
121120
]
122121
ops = {
123122
"add": {

scripts/render_notebooks.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,10 +82,6 @@ def _patched_show(*_args, **_kwargs) -> None:
8282
if not src:
8383
continue
8484

85-
# Progress line bypasses stdout_buf so it survives a crash mid-cell.
86-
preview = src.splitlines()[0][:60]
87-
print(f" [cell {exec_count}] {preview}", file=sys.__stdout__, flush=True)
88-
8985
cell_images.clear()
9086
stdout_buf = io.StringIO()
9187

0 commit comments

Comments
 (0)