Skip to content

cuda: add a cuda-batched runtime that micro-batches concurrent run() …#2421

Open
kali wants to merge 1 commit into
mainfrom
wip/cuda-autobatch
Open

cuda: add a cuda-batched runtime that micro-batches concurrent run() …#2421
kali wants to merge 1 commit into
mainfrom
wip/cuda-autobatch

Conversation

@kali

@kali kali commented Jun 29, 2026

Copy link
Copy Markdown
Collaborator

…calls

Independent caller threads each fire tiny per-token predictor/joiner calls; with many concurrent streams that is one serialized CPU<->GPU round-trip per token per stream. A per-model background worker drains the in-flight queue, concatenates inputs along the symbolic batch axis, runs the inner CUDA runnable once, and scatters the row-slices back to each caller. WIP parked for later.

…calls

Independent caller threads each fire tiny per-token predictor/joiner calls; with many
concurrent streams that is one serialized CPU<->GPU round-trip per token per stream. A
per-model background worker drains the in-flight queue, concatenates inputs along the
symbolic batch axis, runs the inner CUDA runnable once, and scatters the row-slices back
to each caller. WIP parked for later.
@github-actions

Copy link
Copy Markdown

🔴 Bench vs main — 2 speed regression(s)

Reference: main nightly, latest 2026-06-29 (0d old) · PR 6d3dd55e2 · ran on apple-m1-max, i9-11900kb_rtx-4060, jetson-orin-nx · 811 metrics compared

Speed — evaltime · prefill · decode

Δ metric device main → PR
🔴 -6.3% openelm_270M_q40ef16_541
decode · metal
apple-m1-max 237.6 tok/s → 222.6 tok/s
🔴 +6.2% parakeet_tdt_600m_v3_f32f32_preprocessor_1s
evaltime · metal
apple-m1-max 1.49 ms → 1.58 ms
🟢 1 improvement(s)
Δ metric device main → PR
🟢 +3.1% openelm_270M_q40ef16_541
prefill · cpu
apple-m1-max 271.1 tok/s → 279.5 tok/s

lower is better except prefill/decode (tok/s) · adaptive thresholds (max(floor, k×noise) vs the series' own history) · single-shot vs nightly reference · full report → run

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant